Update AlexNet.py