|
CUDA_VISIBLE_DEVICES=0 python experiments/train/train_eval.py \ |
|
'gpus=[0]' \ |
|
overwrite=False \ |
|
resume=True \ |
|
debug=True \ |
|
train.name=rope/train \ |
|
train.source_dataset_name=data/rope_merged/sub_episodes_v \ |
|
train.dataset_name=rope/dataset \ |
|
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=651 \ |
|
train.eval_start_episode=651 \ |
|
train.eval_end_episode=691 \ |
|
train.num_iterations=100000 \ |
|
train.iteration_eval_interval=1000 \ |
|
train.iteration_log_interval=10 \ |
|
train.iteration_save_interval=1000 |
|
|