TomatoBoy90 / AlexnetDemo

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