CUDA_VISIBLE_DEVICES=0,1,2,3 python -m torch.distributed.launch --nproc_per_node=4 --master_port 12346 train_mask_pixel_resnet.py --train_bs 1024 --val_bs 1024 --recon_loss cross_entropy --figname distributed_run_mask_pixel_vae.jpg --runs_name pixelmask_resnet_dist0.01_noise0.01_triplet1 --triplet_weight 1. --dist_weight 0.01 --epoch 100 --drop_prob 0.1 --lr 5e-4 --optim adam --noise_scale 0.01