drn_a_50 missing keys when testing

fyu / drn

Dilated Residual Networks

https://www.vis.xyz/pub/drn

BSD 3-Clause "New" or "Revised" License

1.1k stars 219 forks source link

drn_a_50 missing keys when testing #55

Closed razvancaramalau closed 3 years ago

razvancaramalau commented 3 years ago

Thank you so much for the code! I've encountered an issue when testing the trained DRN-A-50 model. A very long list of missing keys appears when running the segmentation file in test mode. This is by loading the latest checkpoint as pre-trained. Can anyone advise me on this?

razvancaramalau commented 3 years ago

I was not using --resume but it goes cuda out of memory :(