2021 Spring Machine Learning Project (1)
Author by 郭奕玮、郭子奥
Directly run mytrain.py. Change the name of task and model by data_name and model_name. Change the dataset directory by input_root；start_epoch and end_epoch controls the stage of traning. Description can be seen in the code too.
results contains the output of resnet18 and resnet50.
mymodel.py under medmnist is the ResNet model modified by us.
The structure of the code references https://github.com/MedMNIST/MedMNIST
We made some improvements, including thresholds and losses.