train error：Program terminated with signal SIGSEGV

ruoqianguo / DetNet_pytorch

An implementation of DetNet: A Backbone network for Object Detection.

MIT License

377 stars 117 forks source link

train error：Program terminated with signal SIGSEGV #12

Closed DetectionIIT closed 6 years ago

DetectionIIT commented 6 years ago

when /DetNet_Pytorch$ CUDA_VISIBLE_DEVICES=1 python3 trainval_aiia.py exp_name --dataset pascal_voc --net detnet59 --bs 1 --nw 1 --lr 1e-3 --epochs 12 --save_dir weights --cuda --use_tfboard True

I got the error ''Segmentation fault (core dumped)‘’

Please help me. the data in DetNet_Pytorch/data/VOCdevkit2007 is ok?

ruoqianguo commented 6 years ago

This problem is hard to say where is wrong. Maybe GPU out of memory.

DetectionIIT commented 6 years ago

Thanks for you reply.I used pytorch0.2.0 before.now with using pytorch0.3.1 the error disappear.