CCNet
CCNet copied to clipboard
Why is train.py stalled when running?
I have set batch_size = 1 and input_size = 713,713. Use the following command: python train.py --data-dir ./dataset/image/ --random-mirror --random-scale --restore-from ./dataset/resnet101-imagenet.pth --gpu 1,3.6 --learning-rate 1e-2 --input-size 713,713 --weight-decay 1e-4 --batch-size 1 --num-steps 60000 --recurrence 2 --ohem 1 --ohem-thres 0.7 --ohem-keep 100000
However, the following situations occur: UserWarning: size_average and reduce args will be deprecated, please use reduction='elementwise_mean' instead. warnings.warn(warning.format(ret)) 62475 images are loaded!
Then it stuck here!!!!Do you know why? What do I need to do to make it work?