dodox/dppo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix itr initialization in eval agents

Browse Source
This commit is contained in:
allenzren 2024-10-10 14:10:16 -04:00
parent e0842e71dc
commit 4e14b8086d
4 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
agent/eval/eval_diffusion_agent.py
View File

@ -28,7 +28,7 @@ class EvalDiffusionAgent(EvalAgent):
if self.render_video: if self.render_video:
for env_ind in range(self.n_render): for env_ind in range(self.n_render):
options_venv[env_ind]["video_path"] = os.path.join( 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 # Reset env before iteration starts

2
agent/eval/eval_diffusion_img_agent.py
View File

@ -32,7 +32,7 @@ class EvalImgDiffusionAgent(EvalAgent):
if self.render_video: if self.render_video:
for env_ind in range(self.n_render): for env_ind in range(self.n_render):
options_venv[env_ind]["video_path"] = os.path.join( 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 # Reset env before iteration starts

2
agent/eval/eval_gaussian_agent.py
View File

@ -28,7 +28,7 @@ class EvalGaussianAgent(EvalAgent):
if self.render_video: if self.render_video:
for env_ind in range(self.n_render): for env_ind in range(self.n_render):
options_venv[env_ind]["video_path"] = os.path.join( 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 # Reset env before iteration starts

2
agent/eval/eval_gaussian_img_agent.py
View File

@ -32,7 +32,7 @@ class EvalImgGaussianAgent(EvalAgent):
if self.render_video: if self.render_video:
for env_ind in range(self.n_render): for env_ind in range(self.n_render):
options_venv[env_ind]["video_path"] = os.path.join( 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 # Reset env before iteration starts