awaelchliadded this to the v1.9.x milestone 2 years ago
repro
83f41536
move the zero grad call
64631945
awaelchli
changed the title Set grads to None in PyTorch < 2.0 at validation start Set grads to None only once at the beginning of `Trainer.validate/test/predict`2 years ago
awaelchli
changed the title Set grads to None only once at the beginning of `Trainer.validate/test/predict` Fix zero-grad behavior when enterint the validation loop2 years ago
awaelchli
changed the title Fix zero-grad behavior when enterint the validation loop Fix zero-grad behavior when entering the validation loop2 years ago
Login to write a write a comment.
Login via GitHub