Accelerating Deep Learning with Intel Extension for PyTorch: a MedMNIST Classification Decathlon Example
We showcase how to use Intel Extension for PyTorch (IPEX) for training and inference on the MedMNIST datasets, a collection of 10 MNIST-like open datasets on various medical imaging classification tasks such as pathology images, chest x-ray, OCT images. The demo runs on the Intel DevCloud for oneAPI on Ice Lake. We compare the performance with stock PyTorch and observe the performance gain that Intel Extension for PyTorch offers.
Severine Habert
