search

DSGT-DLP / Deep-Learning-Playground

Web Application where people new to Deep Learning can input a dataset and toy around with basic Pytorch modules without writing any code
MIT License
24 stars 8 forks source link

[FEATURE]: Migrate Image Training Endpoint #993

Closed karkir0003 closed 10 months ago

karkir0003 commented 1 year ago

Feature Name

Migrate Image Training Endpoint`

Your Name

Karthik Subramanian

Description

DLP is migrating the training endpoints from Flask Blueprint to Django. This task requires creation of /image route to run image training requests in Django.

References/Resources:

Please ensure that this endpoint is tested on a sample model/architecture for torch built in dataset (eg: MNIST, FashionMNIST, etc) to ensure that the model is training properly

karkir0003 commented 11 months ago

https://github.com/DSGT-DLP/Deep-Learning-Playground/blob/362ee4594c18a2363922b392379fc73388712a9f/backend/common/default_datasets.py#L86-L112

this is the old code for loading image default dataset (it already gives a DataLoader object!!!!)