metadata

title: Assignment for ERA
sdk: gradio
sdk_version: 4.27.0
app_file: app.py
license: mit

SpaceHF: ResNet18 on CIFAR-10

Welcome to the SpaceHF project! This repository contains the implementation of ResNet18 architecture (written in pytorch-lightning) trained on the CIFAR-10 dataset.

Overview

This project aims to demonstrate the training and evaluation of the ResNet18 model on the CIFAR-10 dataset for image classification tasks.

Getting Started

Directory Structure

model.py: resnet18 in lightning.
requirements.txt: List of dependencies.

License

This project is licensed under the MIT License