shan18 / Flash

Create, train and deploy AI models without writing code
https://shan18.github.io/Flash/
MIT License
21 stars 7 forks source link

Add model choices #7

Closed shan18 closed 2 years ago

shan18 commented 2 years ago

In the training module, add more model choices like ResNet18 and other models which can fit into AWS Lambda.

Backend code will be handled by @rvk007 and frontend will be updated by @shan18

shan18 commented 2 years ago

Frontend updated in 7092a5ed5359c04e4d2bef5c42d2a11622fb68bc

rvk007 commented 2 years ago

Completed backed changes in 3dfd3c3f85cc6752959d9d7007b3b2bb27d28586