From 5bf5b25b4e68b606b79fae5c457be4b18f52802a Mon Sep 17 00:00:00 2001 From: atenagm1375 Date: Mon, 29 Mar 2021 15:17:17 +0430 Subject: [PATCH] fix reference variable modification in monitor --- cnsproject/network/monitors.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/cnsproject/network/monitors.py b/cnsproject/network/monitors.py index 4dbef41..8be8a27 100644 --- 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