transformers

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

25 Commits (f46ad5303f91e9947e1d0634776b630f342763bb)