with torch.no_grad() if condition

Bored Coder answered on July 15, 2020 Popularity 6/10 Helpfulness 3/10

answer with torch.no_grad() if condition

related with torch.no_grad()

Comment

  with torch.set_grad_enabled(not no_grad_condition):
    out=network(input)

Popularity 6/10 Helpfulness 3/10 Language python

Tags: python

Link to this answer
Share Copy Link

Contributed on Sep 14 2020

Bored Coder

0 Answers Avg Quality 2/10

Closely Related Answers

Comment

The wrapper "with torch.no_grad()" 
temporarily set all the requires_grad flag to false.

Popularity 10/10 Helpfulness 3/10 Language python

Source: Grepper

Tags: python py

Link to this answer
Share Copy Link

Contributed on Jul 15 2020

Bored Coder

0 Answers Avg Quality 2/10

Continue with Google

By continuing, I agree that I have read and agree to Greppers's Terms of Service and Privacy Policy.