README

project1_model.py contains the PyTorch description of the ResNet model architecture.

project_model.pt contains the trained parameters of the architecture

model_test.ipynb illustrates the whole training process. (It is recommended to use Colab to see the details in the jupyter notebook.)

python requirements

torch

torchvision

torchsummary

torchtoolbox

matplotlib