| 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 |
|
|
