add functionality to Monitor when time is none

Author: Hananel-Hazan

bindsnet/network/monitors.py

@@ -35,8 +35,7 @@ def __init__(
         Constructs a ``Monitor`` object.
 
         :param obj: An object to record state variables from during network simulation.
-        :param state_vars: Iterable of strings indicating names of state variables to
-            record.
+        :param state_vars: Iterable of strings indicating names of state variables to record.
         :param time: If not ``None``, pre-allocate memory for state variable recording.
         :param device: Allow the monitor to be on different device separate from Network device
         """
@@ -48,6 +47,11 @@ def __init__(
         self.batch_size = batch_size
         self.device = device
 
+        # if time is not specified the monitor variable accumulate the logs
+        if self.time is None:
+            self.device = "cpu"
+
+        self.recording = []
         self.reset_state_variables()
 
     def get(self, var: str) -> torch.Tensor:
@@ -56,10 +60,15 @@ def get(self, var: str) -> torch.Tensor:
         Return recording to user.
 
         :param var: State variable recording to return.
-        :return: Tensor of shape ``[time, n_1, ..., n_k]``, where ``[n_1, ..., n_k]`` is
-            the shape of the recorded state variable.
+        :return: Tensor of shape ``[time, n_1, ..., n_k]``, where ``[n_1, ..., n_k]`` is the shape of the recorded state
+        variable.
+        Note, if time == `None`, get return the logs and empty the monitor variable
+
         """
-        return torch.cat(self.recording[var], 0)
+        return_logs = torch.cat(self.recording[var], 0)
+        if self.time is None:
+            self.recording[var] = []
+        return return_logs
 
     def record(self) -> None:
         # language=rst
@@ -83,7 +92,12 @@ def reset_state_variables(self) -> None:
         """
         Resets recordings to empty ``List``s.
         """
-        self.recording = {v: [[] for i in range(self.time)] for v in self.state_vars}
+        if self.time is None:
+            self.recording = {v: [] for v in self.state_vars}
+        else:
+            self.recording = {
+                v: [[] for i in range(self.time)] for v in self.state_vars
+            }
 
 
 class NetworkMonitor(AbstractMonitor):