|
CUDA_VISIBLE_DEVICES=0 python experiments/train/train_eval.py \ |
|
'gpus=[0]' \ |
|
overwrite=False \ |
|
resume=True \ |
|
debug=True \ |
|
train.name=cloth/train \ |
|
train.source_dataset_name=data/cloth_merged/sub_episodes_v \ |
|
train.dataset_name=cloth/dataset \ |
|
sim.preprocess_scale=0.8 \ |
|
sim.num_grippers=2 \ |
|
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=610 \ |
|
train.eval_start_episode=610 \ |
|
train.eval_end_episode=650 \ |
|
train.num_iterations=100000 \ |
|
train.iteration_eval_interval=1000 \ |
|
train.iteration_log_interval=10 \ |
|
train.iteration_save_interval=1000 |
|
|