Currently, only AlexNet on CIFAR dataset is supported. Need to extend support on ImageNet
Currently, only AlexNet on CIFAR dataset is supported. Need to extend support on ImageNet