file structure data.py: data loader model.py: alexnet model train.py: for training test.py: for testing exec python train.py # training python test.py # testing