javiribera / locating-objects-without-bboxes

PyTorch code for "Locating objects without bounding boxes" - Loss function and trained models
Other
249 stars 52 forks source link

About greater batch size during validation #46

Open KMint1819 opened 3 years ago

KMint1819 commented 3 years ago

Hi, thanks for the great work!

When I was training with my own data, the bottleneck of the whole training was apparently the validation process since it's using batch size = 1. In one epoch of my training, it cost only 20% of time for training and back-prop but cost 80% for validation. Why can't the batch size for validation be greater than 1? Is there any issue or problems to be solved in order to have this feature?

Frank-Dz commented 3 years ago

Same question...