resnet-imagenet1k / README.md
nragrawal's picture
Add model weights
86da9f0

ResNet-50 ImageNet Classifier

This model is a ResNet-50 trained on ImageNet dataset.

Model description

  • Model type: ResNet-50
  • Training Data: ImageNet
  • Framework: PyTorch
  • License: MIT

Training procedure

  • Training type: Image Classification
  • Optimizer: SGD
  • Learning rate: 0.1
  • Batch size: 256
  • Number of epochs: 90

Metrics

  • Top-1 Accuracy: XX%
  • Top-5 Accuracy: XX%