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