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
..
2022-05-16 20:52:14 +08:00
2022-04-22 15:54:59 +08:00