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
Download Patch File Download Diff File Expand all files Collapse all files
agent/eval/eval_diffusion_agent.py
+1 -1
View File
@@ -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
agent/eval/eval_diffusion_img_agent.py
+1 -1
View File
@@ -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
agent/eval/eval_gaussian_agent.py
+1 -1
View File
@@ -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
agent/eval/eval_gaussian_img_agent.py
+1 -1
View File
@@ -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