File size: 356 Bytes
404d2af |
1 2 3 4 5 6 7 8 9 10 |
OMP_NUM_THREADS=2 CUDA_VISIBLE_DEVICES='0' python -m torch.distributed.launch --nproc_per_node=1 --master_port 23003 main.py \
--model_name=SGM \
--config_path=configs/sgm.yaml \
--rawdata_path=rawdata \
--desc_path=desc_path \
--desc_suffix=_root_1000.hdf5 \
--dataset_path=dataset_path \
--log_base=log_root_1k_sgm \
--num_kpt=1000 \
--train_iter=900000 |