view reply Could you also remove the --no-tune_diffusion_model, this will tune the entire expert head too. Also you can also use a larger action horizon too by configuring the data_config.py