we use pytorch to build alexnet and train it

`python main.py -mode test`

'python main.py -mode train'

It reduce greatly train time when you use amp,you can try it. python main.py -mode train -amp

Name		Name	Last commit message	Last commit date
Latest commit History 12 Commits
.idea		.idea
__pycache__		__pycache__
data/dogCat		data/dogCat
models		models
README.md		README.md
config.py		config.py
datasets.py		datasets.py
main.py		main.py
test.py		test.py

Provide feedback