/ImageNet

This implements training of popular model architectures, such as AlexNet, ResNet and VGG on the ImageNet dataset(Now we supported alexnet, vgg, resnet, squeezenet, densenet)

Primary LanguagePythonMIT LicenseMIT

Stargazers