cs6910_assignment1

This repo contains solution for assignment1 of Deep Learning Course.
We have attached both cupy and numpy version of our implementation in their respective folders!

Quickstart

Clone the repository: git clone https://github.com/RamanHacks/cs6910_assignment1.git

Navigate to the cupy/numpy folder: cd numpy #cupy

Install the requirements: pip install -r requirements.txt

Run training script: python train_fmnist.py

For detailed instructions, read these docs: numpy, cupy