Remove manual torch.no_grad
and train eval mode in LightningModule
#1100
+31
−39
torch.no_grad
and train eval mode in LightningModule
#1100