search

leeyeehoo / CSRNet-pytorch

CSRNet: Dilated Convolutional Neural Networks for Understanding the Highly Congested Scenes
642 stars 259 forks source link

training error #24

Open liuleiBUAA opened 5 years ago

liuleiBUAA commented 5 years ago

When I train the model, the loss is nan Epoch: [0][1170/1200] Time 0.709 (0.417) Data 0.020 (0.017) Loss nan (nan)

leeyeehoo commented 5 years ago

Yes. You can make learning rate smaller. like 1e-8 .