Log n_actors

laboroai · Aug 25, 2024 · 0589202 · 0589202
1 parent 7b5d843
commit 0589202
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/border/examples/mujoco/sac_mujoco_async_tch.rs b/border/examples/mujoco/sac_mujoco_async_tch.rs
@@ -221,6 +221,7 @@ mod utils {
                 recorder_run.set_tag("env", env_name)?;
                 recorder_run.set_tag("algo", "sac")?;
                 recorder_run.set_tag("backend", "tch")?;
+                recorder_run.set_tag("n_actors", args.n_actors.to_string())?;
                 Ok(Box::new(recorder_run))
             }
             false => Ok(Box::new(TensorboardRecorder::new(model_dir))),