model
model copied to clipboard
Add scripts to export the encoder of Clay
- [x] Add code to take the clay encoder & compile it to torch exported program format
- [x] Add code to compile torch encoder into onnx format
- [x] Notebook to show how to use the compiled models
Note: onnx gpu export is not implemented, will fix that in a new PR.