Closed shan18 closed 2 years ago
While downloading the trained models, the user should have the option to download the model in PyTorch, ONNX or TensorFlow format.
Currently support for only PyTorch and ONNX will be provided.
Training server-side ONNX conversion done in #15
While downloading the trained models, the user should have the option to download the model in PyTorch, ONNX or TensorFlow format.