fix reference variable modification in monitor

cnrl · Mar 29, 2021 · 5bf5b25 · 5bf5b25
1 parent 47b9376
commit 5bf5b25
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/cnsproject/network/monitors.py b/cnsproject/network/monitors.py
@@ -92,8 +92,7 @@ def record(self) -> None:
 
         """
         for var in self.state_variables:
-            data = getattr(self.obj, var)
-            data.unsqueeze_(0)
+            data = getattr(self.obj, var).unsqueeze(0)
             self.recording[var].append(
                 torch.empty_like(data, device=self.device).copy_(
                     data, non_blocking=True