AlexNet parameters?

Hi all,

I am training the AlexNet model using the example/image-classification/train_imagenet.py script on a single node, using one GPU. Has anyone found parameters that give good performance on this model?

Thanks in advance,
Kevin

This is the training script for many possible models. Which model are you training? The default is resnet-50.

I remember using it a long time ago and it was good enough!

Hi,

I have checked that I am indeed training AlexNet.

Kevin