search

microsoft / onnxruntime-training-examples

Examples for using ONNX Runtime for model training.
MIT License
311 stars 62 forks source link

[Request] Need a example for training an ONNX model #58

Closed fumihwh closed 2 years ago

fumihwh commented 3 years ago

As title says, if any ONNX model sample is available.....

liqunfu commented 3 years ago

https://github.com/microsoft/onnxruntime/blob/master/orttraining/orttraining/test/python/orttraining_test_ortmodule_bert_classifier.py

See instructions in the same folder as well.

fumihwh commented 3 years ago

@liqunfu Thanks for reply. I notice the file https://github.com/microsoft/onnxruntime/blob/d8cf9604123e078b2f12e7c13902ba3a0ae56244/orttraining/orttraining/test/python/orttraining_test_orttrainer_bert_toy_onnx.py#L85 but I can't find the model file testdata/bert_toy_postprocessed.onnx, also any script or shell which can download it.

priumoraes commented 3 years ago

Once you follow the recipe, the .onnx file will be generated.

askhade commented 2 years ago

Closing as old. Please reopen a new issue if this is not yet resolved. Thanks!