Open Shruthi42 opened 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
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?
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