torch
torch copied to clipboard
Persist `training` attribute
This should fix #632 but it seems that https://github.com/pytorch/pytorch/issues/62603 is the reson why it doesn't work yet.