onnxruntime
Add new PytTrch front-end
#4815
Merged
Overview
Commits
27
Changes
View On GitHub

Add new PytTrch front-end #4815

thiagocrepaldi merged 27 commits into master from feature/new_pytorch_frontend
thiagocrepaldi
thiagocrepaldi thiagocrepaldi added component:training-frontend
thiagocrepaldi thiagocrepaldi requested a review from liqunfu liqunfu 5 years ago
thiagocrepaldi thiagocrepaldi requested a review from BowenBao BowenBao 5 years ago
thiagocrepaldi thiagocrepaldi requested a review from rayankrish rayankrish 5 years ago
thiagocrepaldi thiagocrepaldi requested a review from spandantiwari spandantiwari 5 years ago
thiagocrepaldi thiagocrepaldi requested a review 5 years ago
thiagocrepaldi thiagocrepaldi assigned thiagocrepaldi thiagocrepaldi 5 years ago
thiagocrepaldi thiagocrepaldi force pushed 5 years ago
thiagocrepaldi thiagocrepaldi force pushed 5 years ago
thiagocrepaldi thiagocrepaldi force pushed to c7e152e0 5 years ago
thiagocrepaldi thiagocrepaldi force pushed 5 years ago
thiagocrepaldi thiagocrepaldi force pushed 5 years ago
thiagocrepaldi thiagocrepaldi force pushed 5 years ago
Add ORTTrainerOptions class for the new pytorch frontend (#4382)
e78c2530
Add _ORTTrainerModelDesc to perform validation for model description …
0203d130
Add Loss Scaler classes to the new frontend (#4306)
a6c77033
Add TrainStepInfo used on the new frontend API (#4256)
b9b3b08a
Add Optimizer classes to the new frontend (#4280)
da8b45b7
Add LRScheduler implementation (#4357)
afcdc57d
Add basic ORTTrainer API (#4435)
04e11b1b
Add opset_version into ORTTrainerOptions and change type of ORTTraine…
fafd801f
Update ModelDescription and minor fix on ORTTrainer ctor (#4605)
8466160e
rayankrish Add ONNX Model Export to New Frontend (#4612)
ac3cba14
Create training session + minor improvements (#4668)
da58d070
Save ONNX model in file (#4671)
88f0dbfc
Add eval step (#4674)
951ef089
Add train_step (#4677)
d55290a4
rayankrish Add LR Scheduler (#4694)
e8269a45
rayankrish Add deterministic compute tests (#4716)
1cbe9a3b
rayankrish Add legacy vs experimental ORTTrainer accuracy comparison (#4727)
110609f5
Add Mixed precision/LossScaler + several fixes (#4739)
87d57b25
rayankrish Add ONNX BERT example + minor fixes (#4757)
4568982e
Add Dynamic Shape support (#4758)
0f786f82
Update DeepSpeed Zero Stage option to a separate option group (#4772)
f72d1f3f
Add support to fetches (#4777)
b07e69bd
Add Gradient Accumulation Steps support (#4793)
a538222b
Fix Dynamic Axes feature and add unit test (#4795)
16ea05fd
Add frozen weights test (#4807)
34de2061
Move new pytorch front-end to 'experimental' namespace (#4814)
35bc8139
Fix build
af5106f6
thiagocrepaldi thiagocrepaldi force pushed to af5106f6 5 years ago
spandantiwari
spandantiwari
spandantiwari commented on 2020-08-15
spandantiwari
spandantiwari approved these changes on 2020-08-15
thiagocrepaldi thiagocrepaldi merged 42408aa3 into master 5 years ago
thiagocrepaldi thiagocrepaldi deleted the feature/new_pytorch_frontend branch 5 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum