Mini-Project 1: Residual Network Design

Environment

python 3.7.12

torch 1.10.0+cu111+cudnn8805

How to run

Train

python project1_model.py -train -epoch 1000

Test

python project1_model.py -test

Summary

python project1_model.py -summary

If you encounter any troubles while running the .py file, please use the .ipynb file instead.

In the .ipynb file:

train==1 --> train mode

train==0 --> test mode