minors

justanhduc · Jan 11, 2022 · 46fc4b4 · 46fc4b4
1 parent c629664
commit 46fc4b4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/neural_monitor/monitor.py b/neural_monitor/monitor.py
@@ -604,7 +604,7 @@ def iter_epoch(self, iterator: Iterable) -> Any:
         Examples
         --------
 
-        >>> from neuralnet_pytorch import monitor as mon
+        >>> from neural_monitor import monitor as mon
         >>> mon.print_freq = 1000
         >>> num_epochs = 10
         >>> for epoch in mon.iter_epoch(range(mon.epoch, num_epochs))
@@ -638,7 +638,7 @@ def iter_batch(self, iterator: Iterable) -> Any:
         Examples
         --------
 
-        >>> from neuralnet_pytorch import monitor as mon
+        >>> from neural_monitor import monitor as mon
         >>> mon.print_freq = 1000
         >>> data_loader = ...
         >>> num_epochs = 10
@@ -1726,6 +1726,7 @@ def post_hook(mod, _inputs, outputs):
         hooks += [mod.register_forward_hook(post_hook) for mod in module.modules()]
 
         # Run module.
+        module.eval()
         module(*inputs)
         for hook in hooks:
             hook.remove()