Xueyun Zhu
f1ba9aaf34
Add pipeline transformer for wait/record node ( #3513 )
...
* pipeline transformer
* clean up
* address feedback
* add record/wait for first stage and updated split script
* address feedback
* make recv/send signal as initializer
* merge
* address feedback
* unify input and initializer
* address feedback and bug fix
* minor fix
* windows build
* fix
2020-04-22 23:28:01 -07:00
pengwa
2c7c45076b
MaxBatchSize E2E Test ( #3454 )
...
* max batch size e2e test
*update test data snapshot
2020-04-15 09:50:44 +08:00
Thiago Crepaldi
15e32b44fd
Merge pull request #3383
...
Merge from master into ort_training
2020-04-06 19:05:01 -07:00
Edward Chen
95707d22a5
Disable gradient clipping for E2E test.
2020-04-06 23:07:28 +00:00
Xueyun Zhu
efc8bd738f
add pipeline graph split script ( #3275 )
...
* pipeline graph cut
* add element type
* add input wait event and shape info
* shape inference
* support multiple cuts
* format script
* address feedback
* address feedback
2020-03-31 19:30:18 -07:00
edgchen1
d9f628cb1d
Remove orttraining/tools/scripts/profile directory. ( #3268 )
2020-03-19 14:13:05 -07:00
Jesse Benson
3a7539e071
Update bert-base convergence values
2020-03-13 23:03:34 -07:00
Edward Chen
e542cfd0e0
Introduce training changes.
2020-03-11 14:39:03 -07:00