Skip to content

Commit 5bf5b25

Browse files
committed
fix reference variable modification in monitor
1 parent 47b9376 commit 5bf5b25

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

cnsproject/network/monitors.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -92,8 +92,7 @@ def record(self) -> None:
9292
9393
"""
9494
for var in self.state_variables:
95-
data = getattr(self.obj, var)
96-
data.unsqueeze_(0)
95+
data = getattr(self.obj, var).unsqueeze(0)
9796
self.recording[var].append(
9897
torch.empty_like(data, device=self.device).copy_(
9998
data, non_blocking=True

0 commit comments

Comments
 (0)