search

microsoft / InnerEye-DeepLearning

Medical Imaging Deep Learning library to train and deploy 3D segmentation models on Azure Machine Learning
https://aka.ms/innereyeoss
MIT License
553 stars 142 forks source link

Allow batch sizes > 1 for classification model inference. #453

Open Shruthi42 opened 3 years ago

Shruthi42 commented 3 years ago

At the moment, we set batch size =1 in the dataloaders when running inference for a classification model.

https://github.com/microsoft/InnerEye-DeepLearning/blob/daefdba6083775de7ca258d18ae315e57bcb54bd/InnerEye/ML/model_testing.py#L428

AB#3998

hxri commented 3 years ago

At the moment, we set batch size =1 in the dataloaders when running inference for a classification model.

https://github.com/microsoft/InnerEye-DeepLearning/blob/daefdba6083775de7ca258d18ae315e57bcb54bd/InnerEye/ML/model_testing.py#L428

AB#3998

Hi Can I work on this?