/pytorch_imagenet

PyTorch models for imagenet classification

Primary LanguagePythonMIT LicenseMIT

PyTorch Image Classification

  • PyTorch models for imagenet classification.
  • These models can be utilized as pre-training models for other computer vision tasks.
  • Training with the nets provided by models/net_factory.py(self-define & in torch models)

Training results

Model Top-1 error (%) Top-5 error (%) Links Notes
PVANet 26.92 8.84 Model download / Acc. plot / Loss plot 600 epochs

References