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)
I clone the project and reexecute it with resnet50. However, it comes out only Prec@1 72.780 Prec@5 90.880. I only add "nn.DataParallel". Could you give me the final *.pth of resnet50?
There may be something wrong in top5 accuracy updating, which making the top5 accuracy is the same as the top1.
The code with problem(main.py) is as follow: