search

andife / power_spherical

Pytorch implementation of the Power Spherical distribution
https://arxiv.org/abs/2006.04437
MIT License
0 stars 3 forks source link

WIP: Fix onnx export #4

Closed andife closed 2 months ago

andife commented 3 months ago

try to fix the onnx export