How to train this model?

leeyeehoo / CSRNet-pytorch

CSRNet: Dilated Convolutional Neural Networks for Understanding the Highly Congested Scenes

642 stars 259 forks source link

How to train this model? #26

Closed LaiPiXiong closed 5 years ago

LaiPiXiong commented 5 years ago

How to train this model? I use your pretrained parameters, it can get good result. But, not all Fine-Tunning parameters are released. I want to train this model on MALL dataset. BUT, the evaluate always same in different epochs. MY batch_size is 1, lr is 0.001, optimizer is SGD. Can you give me some advice about how to train this model? @leeyeehoo