diff --git a/agent/eval/eval_diffusion_agent.py b/agent/eval/eval_diffusion_agent.py index 00ffab5..577c11b 100644 --- a/agent/eval/eval_diffusion_agent.py +++ b/agent/eval/eval_diffusion_agent.py @@ -28,7 +28,7 @@ class EvalDiffusionAgent(EvalAgent): if self.render_video: for env_ind in range(self.n_render): options_venv[env_ind]["video_path"] = os.path.join( - self.render_dir, f"itr-{self.itr}_trial-{env_ind}.mp4" + self.render_dir, f"eval_trial-{env_ind}.mp4" ) # Reset env before iteration starts @@ -62,7 +62,7 @@ class EvalDiffusionAgent(EvalAgent): ) reward_trajs[step] = reward_venv firsts_trajs[step + 1] = terminated_venv | truncated_venv - + # update for next step prev_obs_venv = obs_venv diff --git a/agent/eval/eval_diffusion_img_agent.py b/agent/eval/eval_diffusion_img_agent.py index 6002413..d3b7bf2 100644 --- a/agent/eval/eval_diffusion_img_agent.py +++ b/agent/eval/eval_diffusion_img_agent.py @@ -32,7 +32,7 @@ class EvalImgDiffusionAgent(EvalAgent): if self.render_video: for env_ind in range(self.n_render): options_venv[env_ind]["video_path"] = os.path.join( - self.render_dir, f"itr-{self.itr}_trial-{env_ind}.mp4" + self.render_dir, f"eval_trial-{env_ind}.mp4" ) # Reset env before iteration starts @@ -65,7 +65,7 @@ class EvalImgDiffusionAgent(EvalAgent): ) reward_trajs[step] = reward_venv firsts_trajs[step + 1] = terminated_venv | truncated_venv - + # update for next step prev_obs_venv = obs_venv diff --git a/agent/eval/eval_gaussian_agent.py b/agent/eval/eval_gaussian_agent.py index 96b304d..af9245c 100644 --- a/agent/eval/eval_gaussian_agent.py +++ b/agent/eval/eval_gaussian_agent.py @@ -28,7 +28,7 @@ class EvalGaussianAgent(EvalAgent): if self.render_video: for env_ind in range(self.n_render): options_venv[env_ind]["video_path"] = os.path.join( - self.render_dir, f"itr-{self.itr}_trial-{env_ind}.mp4" + self.render_dir, f"eval_trial-{env_ind}.mp4" ) # Reset env before iteration starts diff --git a/agent/eval/eval_gaussian_img_agent.py b/agent/eval/eval_gaussian_img_agent.py index 2a8e54a..e607cba 100644 --- a/agent/eval/eval_gaussian_img_agent.py +++ b/agent/eval/eval_gaussian_img_agent.py @@ -32,7 +32,7 @@ class EvalImgGaussianAgent(EvalAgent): if self.render_video: for env_ind in range(self.n_render): options_venv[env_ind]["video_path"] = os.path.join( - self.render_dir, f"itr-{self.itr}_trial-{env_ind}.mp4" + self.render_dir, f"eval_trial-{env_ind}.mp4" ) # Reset env before iteration starts