Face and head loss are NAN, when training PyramidBox. Online Urgent!
Created by: YueXiNPU
Hello everyone and @qingqing01. I am training PyramidBox (ECCV 2018) by Baidu with 3 GPUs of GTX 1080Ti, memory 11G. Training command is python -u train.py --batch_size=3. I try batch_size=4/5/6, but it does not work for me. Nan is shown as below: GPU information is shown as below: Anyone can help me to solve this problem?