EfficientNet-PyTorch icon indicating copy to clipboard operation
EfficientNet-PyTorch copied to clipboard

Published 20 hours ago • verified publisher icon lukemelas

inference a batch images in one time

Open simplew2011 opened this issue 4 years ago • 2 comments

how can i using efficientnet to inference a batch images in one time, not circle inference in every image.

simplew2011 avatar Dec 19 '20 04:12 simplew2011

What do you mean? Something different from normal pytorch batching?

lukemelas avatar Dec 21 '20 18:12 lukemelas

If you use Dataloader, the simplest way is to change the batch_size to the size of your dataset. Be sure that you have enough memory to do that.

Surayuth avatar Dec 22 '20 12:12 Surayuth