PyTorch Forums
Does model.eval() & with torch.set_grad_enabled(is_train) have the same effect for grad history?
autograd
SimonW
(Simon Wang)
April 29, 2018, 4:27am
3
eval doesn’t turn off history tracking.
4 Likes
show post in topic