model.eval() when testing

cvlab-stonybrook / DM-Count

Code for NeurIPS 2020 paper: Distribution Matching for Crowd Counting.

MIT License

218 stars 52 forks source link

model.eval() when testing #3

Closed enric1994 closed 4 years ago

enric1994 commented 4 years ago

I think you should add model.eval() in test.py. It won't alter the default result, but if batch normalization is enabled it will definitely affect the result. I had to change it because I am using ResNet50 instead of VGG19.

Boyu-Wang commented 4 years ago

Thanks. Updated.