Jingyu Kong

icon company Tianjin University icon location I am Jingyu Kong, a student in Tianjin University. I want to study in Github and try to make some contributions to the community, I like coding.

Results 1 comments of Jingyu Kong

How to train on multi GPU?

When I run: python scripts/segmentation_train.py --data_name PROMISE12 --image_size 256 --num_channels 128 --class_cond False --num_res_blocks 2 --num_heads 1 --learn_sigma True --use_scale_shift_norm False --attention_resolutions 16 --diffusion_steps 1000 --noise_schedule linear --rescale_learned_sigmas False --rescale_timesteps...