训练MedSegDiff时出现Connection refused
解决方法
先开一个终端,将8850端口起起来
python -m visdom.server -port 8850
在运行训练代码
python scripts/segmentation_train.py --data_dir data/training --out_dir output --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 False --lr 1e-4 --batch_size 16
即可正常训练