2 Commits

Author SHA1 Message Date
pkufool
90bc61e970 Fix tests 2022-06-27 19:46:55 +08:00
Fangjun Kuang
f6ce135608
Various fixes to support torch script. (#371)
* Various fixes to support torch script.

* Add tests to ensure that the model is torch scriptable.

* Update tests.
2022-05-16 21:46:59 +08:00