fix itr initialization in eval agents
This commit is contained in:
@@ -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
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user