we use pytorch to build alexnet and train it 1.test use trained model `python main.py -mode test` 2.train dog/cat dataset 'python main.py -mode train' 3.train model use amp It reduce greatly train time when you use amp,you can try it. python main.py -mode train -amp