logo

28 Commits (0592e151a04c8b4ef93c8a9c5ac6d26b119ec104)

Author SHA1 Message Date
Jael Gu 0d30fc46a7 Add model 2 years ago
ChengZi ea58513a8f add relative package import 2 years ago
Jael Gu a00786d445 Update parameters 2 years ago
ChengZi f46ad5303f train mlm and clm task 2 years ago
Jael Gu a307a3196a Update model list 2 years ago
Jael Gu d838ba8357 Replace with transformers.onnx 2 years ago
Jael Gu 6d228847cf Update save_onnx 2 years ago
Jael Gu 9acb9072e1 Fix log 2 years ago
Jael Gu 1377e6a08e Fix issue due to enable_onnx_checker only available in torch 0.9 2 years ago
Jael Gu f0131944b4 Add log to test_onnx 2 years ago
Jael Gu 73fc63b013 Update for onnx test 2 years ago
Jael Gu 908e07a327 Add device 2 years ago
Jael Gu f5865f2021 Update model list for onnx 2 years ago
Jael Gu 976368836d Update todo 3 years ago
Jael Gu c6eabd756e Add list of supported_model_names(format='onnx') 3 years ago
Jael Gu a55e58b29b Update 3 years ago
Jael Gu c1351df718 Update 3 years ago
Jael Gu 516372f339 Allow to save model as onnx 3 years ago
Jael Gu 7082875b8b Debug 3 years ago
Jael Gu 2bc56f2cd6 Add supported_model_names 3 years ago
Jael Gu 71c4b9e93a Debug 3 years ago
Jael Gu 0bd1bcc127 Update model list 3 years ago
Jael Gu 0ee984c935 Modify save_model 3 years ago
Jael Gu ebfc9c74e3 Update 3 years ago
Jael Gu fd966f51bd Support save model with torchscript 3 years ago
Jael Gu 671ed01d78 Update 3 years ago
Jael Gu 1389d9923c Refactor 3 years ago
Jael Gu 923b3c6ff4 Refactor operator 3 years ago