search

multimodallearning / pytorch-mask-rcnn

Other
2.01k stars 557 forks source link

Inference on validation step #86

Open evinpinar opened 5 years ago

evinpinar commented 5 years ago

During validation epoch (valid_epoch), the prediction is made by passing the "training" argument. here And this sets model to training mode. Here, shouldn't we use the "inference" setting and set the model to val() mode?

Thanks