README.md

metadata

title: Custom Resnet Cifar10 Classifier
emoji: 🏃
colorFrom: indigo
colorTo: gray
sdk: gradio
sdk_version: 3.39.0
app_file: app.py
pinned: false
license: mit

Train the Custom ResNet Model using Pytorch Lighning

Using Collab train the Custom ResNet Model using Pytorch Lightning

Download the Model weights

Download the model weights

Create account in Hugging Face

Create a user account in Hugging Face

Create a new space

Create a new space using URL: https://huggingface.co/new-space

Space Name: custom_resnet_cifar10_classifier License: MIT

Upload files

• Upload requirements.txt file: with pytorch-lightning torch torchvision torch-lr-finder grad-cam pillow numpy

• Upload 10 example image files:

• Upload the model weights file

• Create app.py with the following logic

  • Load the weights
  • Logic for inference
  • Gradio application with the parameters specified

Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference