pgnd / src /experiments /scripts /train_cloth.sh
kywind
update
f96995c
raw
history blame contribute delete
661 Bytes
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