Can “set_grad_enabled” replace “no_grad” and “enable_grade”?
set_grad_enabled can be used as a function or context manager as an alternative for torch.no_grad and torch.enable_grad.
1 Like
Thank you very much for your reply!