search

shan18 / Flash

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

Option for Transfer Learning #6

Closed shan18 closed 2 years ago

shan18 commented 2 years ago

While training the model, give the user an option to pre-train their models on the following datasets:

This milestone will be done in two steps:

  1. Updating the training server code (the cifar100 model checkpoints will be present in the EC2 instance itself) [Handled by @rvk007]
  2. Showing the configuration options in the frontend [Handled by @shan18]
shan18 commented 2 years ago

Changes from frontend and lambda completed in b11579e9f77bb34ad3c323ebce2ac2e49f2da881

rvk007 commented 2 years ago

Completed changes from backend for transfer learning in 9ed8b10b6014cc5252c22fec6b282fd5ecbdd9fb