removed dead code in grads

2026-09-09 11:32:07 +08:00 · 2019-07-24 16:19:19 -04:00
parent 97aa69c8f6
commit f3b0cbf998
2 changed files with 10 additions and 4 deletions
@@ -21,6 +21,7 @@ class LightningModule(GradInformation, ModelIO, ModelHooks):
        self.loaded_optimizer_states_dict = {}
        self.trainer = None
        self.experiment = None
+        self.example_input_array = None

        # track if gpu was requested for checkpointing
        self.on_gpu = False