| 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 |