fix gradient logging (remove hook if trainer does not exist anymore)

inferno-pytorch · Aug 9, 2019 · cd4de05 · cd4de05
1 parent bd63892
commit cd4de05
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/inferno/trainers/callbacks/gradients.py b/inferno/trainers/callbacks/gradients.py
@@ -24,6 +24,12 @@ def log_every(self, value):
                 FrequencyValueError)
 
     def hook(self, module, grad_input, grad_output):
+
+        #remove hook if trainer does not exits
+        if self.trainer is None:
+            self.hook_handle.remove()
+            return
+
         if self.log_every.match(iteration_count=self.trainer.iteration_count,
                                 epoch_count=self.trainer.epoch_count,
                                 persistent=True, match_zero=True):