| CUDA_VISIBLE_DEVICES=0 python experiments/train/train_eval.py \ |
| 'gpus=[0]' \ |
| overwrite=False \ |
| resume=True \ |
| debug=True \ |
| train.name=box/train \ |
| train.source_dataset_name=data/box_merged/sub_episodes_v \ |
| train.dataset_name=box/dataset \ |
| sim.num_grippers=2 \ |
| sim.gripper_points=True \ |
| sim.gripper_forcing=False \ |
| model.material.radius=0.0 \ |
| train.dataset_load_skip_frame=3 \ |
| train.dataset_skip_frame=1 \ |
| train.num_workers=8 \ |
| train.batch_size=32 \ |
| train.training_start_episode=0 \ |
| train.training_end_episode=303 \ |
| train.eval_start_episode=303 \ |
| train.eval_end_episode=323 \ |
| train.num_iterations=100000 \ |
| train.iteration_eval_interval=1000 \ |
| train.iteration_log_interval=10 \ |
| train.iteration_save_interval=1000 |
|
|
