model icon indicating copy to clipboard operation
model copied to clipboard

Add scripts to export the encoder of Clay

Open srmsoumya opened this issue 7 months ago • 2 comments

  • [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.

srmsoumya avatar Jul 18 '24 12:07 srmsoumya