shan18 / Flash

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

Cross-Platform Downloads #11

Closed shan18 closed 2 years ago

shan18 commented 2 years ago

While downloading the trained models, the user should have the option to download the model in PyTorch, ONNX or TensorFlow format.

shan18 commented 2 years ago

Currently support for only PyTorch and ONNX will be provided.

rvk007 commented 2 years ago

Training server-side ONNX conversion done in #15