updated args

2026-09-09 11:32:07 +08:00 · 2019-06-25 19:18:27 -04:00
parent 89410e9090
commit e801914d1d
1 changed files with 1 additions and 0 deletions
@@ -163,6 +163,7 @@ class Trainer(TrainerIO):
        # enable eval mode
        model.zero_grad()
        model.eval()
+        model.from_lightning = True

        # disable gradients to save memory
        torch.set_grad_enabled(False)