AlexNet parameters?


#1

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


#2

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


#3

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


#4

Hi,

I have checked that I am indeed training AlexNet.

Kevin