Hzzone / AgeFlow

AgeFlow: Conditional Age Progression and Regression with Normalizing Flows (IJCAI 2021)
20 stars 3 forks source link

RuntimeError #9

Open quleileileilei opened 1 year ago

quleileileilei commented 1 year ago

when I run python train.py --batch_size 16 --max_iter 1000000 --n_flow 32 --n_block 4 --n_bits 5 --lr 1e-4 --img_size 64 --temp 0.7 --gradient_accumulation_steps 1 --checkpoint_iter 100000 --generation_iter 100 --dataset_name FFHQ --network_config 1 --cls_loss_weight 0.001

I meet a error “ RuntimeError: Default process group has not been initialized, please make sure to call init_process_group.“

Can anyone help me?