| CUDA_VISIBLE_DEVICES=0 python experiments/train/train_eval.py \ | |
| 'gpus=[0]' \ | |
| overwrite=False \ | |
| resume=True \ | |
| debug=True \ | |
| train.name=bread/train \ | |
| train.source_dataset_name=data/bread_merged/sub_episodes_v \ | |
| train.dataset_name=bread/dataset \ | |
| sim.preprocess_scale=3.0 \ | |
| sim.num_grippers=2 \ | |
| sim.n_particles=500 \ | |
| 'sim.num_grids=[100, 100, 100, 0.01]' \ | |
| model.gripper_radius=0.12 \ | |
| model.material.radius=0.0 \ | |
| model.material.pe_num_func_res=4 \ | |
| train.dataset_load_skip_frame=3 \ | |
| train.dataset_skip_frame=1 \ | |
| train.num_workers=8 \ | |
| train.batch_size=16 \ | |
| train.training_start_episode=0 \ | |
| train.training_end_episode=143 \ | |
| train.eval_start_episode=143 \ | |
| train.eval_end_episode=163 \ | |
| train.num_iterations=100000 \ | |
| train.iteration_eval_interval=1000 \ | |
| train.iteration_log_interval=10 \ | |
| train.iteration_save_interval=1000 | |