Justin Queiros de Oliveira

@woltium Hello, How did Diffusion Policy even run for you? Even after following all the instructions I get an error until I turn off ENABLE_EMA = True in policy.py to...

@woltium I used this command: python3 imitate_episodes.py --task_name sim_transfer_cube_scripted --ckpt_dir /home/Desktop/learning --policy_class Diffusion --chunk_size 32 --batch_size 32 --lr 1e-5 --seed 0 --num_steps 100000 --eval_every 2000 --validate_every 2000 --save_every 2000

@barsm42 have you found a solution to this? I am searching for one myself, I encountered the same problem

@barsm42 I found it, it works but the parameters need to be set differently. try this: python3 imitate_episodes.py \ --task_name aloha_mobile_wipe_wine \ --ckpt_dir /scr/tonyzhao/train_logs/wipe_wine_diffusion_seed0 \ --policy_class Diffusion --chunk_size 32 \...

can you post a screenshot? Those arguments are definitely part of the imitate_episode.py script, look at this section "def main(args): set_seed(1) # command line parameters is_eval = args['eval'] ckpt_dir =...