lllukehuang/EE228-Mini-Project-MedMNIST

EE228-Mini-Project: Medical Image Classification on MedMNIST

The main code structure is referred to MedMNIST.

How to Run Our codes:

1. Create environment first

conda create -n ML python=3.6
conda activate ML
conda install PyTorch==0.3.1
conda install numpy==1.18.5, pandas==0.25.3, scikit-learn==0.22.2, tqdm

2. Run the shell script.

./run.sh

Experiment Environment

• Ubuntu16.04
• NVIDIA GTX1080TI

Video memory overhead:

Resnet-18: 1777MB

Resnet-50: 4977MB