|
--- |
|
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 |
