eval_episodes=1 env.episode_length=8

This commit is contained in:
Cadene 2024-04-10 14:44:04 +00:00
parent 48ec479660
commit 79542ecd13
1 changed files with 4 additions and 0 deletions

View File

@ -142,6 +142,7 @@ jobs:
wandb.enable=False \ wandb.enable=False \
offline_steps=2 \ offline_steps=2 \
online_steps=0 \ online_steps=0 \
eval_episodes=1 \
device=cpu \ device=cpu \
save_model=true \ save_model=true \
save_freq=2 \ save_freq=2 \
@ -179,6 +180,7 @@ jobs:
wandb.enable=False \ wandb.enable=False \
offline_steps=2 \ offline_steps=2 \
online_steps=0 \ online_steps=0 \
eval_episodes=1 \
device=cpu \ device=cpu \
save_model=true \ save_model=true \
save_freq=2 \ save_freq=2 \
@ -202,6 +204,7 @@ jobs:
policy=diffusion \ policy=diffusion \
env=pusht \ env=pusht \
eval_episodes=1 \ eval_episodes=1 \
env.episode_length=8 \
device=cpu device=cpu
- name: Test train TDMPC on Simxarm end-to-end - name: Test train TDMPC on Simxarm end-to-end
@ -213,6 +216,7 @@ jobs:
wandb.enable=False \ wandb.enable=False \
offline_steps=1 \ offline_steps=1 \
online_steps=1 \ online_steps=1 \
eval_episodes=1 \
device=cpu \ device=cpu \
save_model=true \ save_model=true \
save_freq=2 \ save_freq=2 \