fix

Author: FedeClaudi

proj/environment/manager.py

@@ -107,7 +107,7 @@ def _save_video(self):
             animate_from_images(
                 str(self.frames_folder),
                 str(self.datafolder / f"{self.exp_name}.mp4"),
-                int(round(1 / self.mouse.dt)),
+                int(round(1 / self.model.dt)),
             )
         except (ValueError, FileNotFoundError):
             print("Failed to generate video from frames.. ")

proj/run/runner.py

@@ -38,7 +38,7 @@ def render(self, task):
 
 # run
 def run_experiment(
-    environment, controller, model, n_secs=0.5, frames_folder=None,
+    environment, controller, model, n_secs=8, frames_folder=None,
 ):
     """
         Runs an experiment

rnn/cartesian_rnn.py

@@ -85,12 +85,15 @@ def __getitem__(self, idx):
             # Get a simulation folder and load data
             fld = random.choice(self.simulation_folders)
 
-            (
-                config,
-                trajectory,
-                history,
-                cost_history,
-            ) = load_results_from_folder(fld)
+            try:
+                (
+                    config,
+                    trajectory,
+                    history,
+                    cost_history,
+                ) = load_results_from_folder(fld)
+            except ValueError:
+                continue
 
             # Get controls history
             controls = np.vstack(history[["tau_r", "tau_l"]].values)
@@ -150,7 +153,8 @@ def init_hidden(self,):
 
     def forward(self, X):
         # Reshape X: n_steps X batch_size X n_inputs
-        X = X.unsqueeze(0)
+        # X = X.unsqueeze(0)
+        X = X.permute((1, 0, 2))
 
         # for each time step
         self.hidden = self.rnn(X, self.hidden)

run_allocentric.py

@@ -25,6 +25,7 @@
 # plot_trajectory(env.reset())
 
 # %%
-run_experiment(env, control, model)
+run_experiment(env, control, model, n_secs=0.1)
+
 
 # %%