Commit Graph

8 Commits (11ef05ca32f55cc9e9f07f30ddec5c8a32eec4ef)

Author SHA1 Message Date
littletomatodonkey 8fd56a4503
fix static train (#478) 2020-12-15 16:43:15 +08:00
huangxu96 dc3020ab4a
support fp16 training (#435)
* support fp16 training

* Use compiled training program

* Change timing ips.

* Use dali

* add pure fp16 training

* fix a bug, which will not use fuse pass using pure fp16 training.

* modify code as review

* modify loss, so that it will use different loss when using pure fp16 training.

* remove some fluid API

* add static optimizer.
2020-12-11 11:04:51 +08:00
QingshuChen 066d53f8ec
support cpu/xpu/gpu in static graph (#460) 2020-12-08 20:59:23 +08:00
littletomatodonkey e92cb0b93c
fix init model in static mode (#444) 2020-12-03 12:50:33 +08:00
Tingquan Gao 2b77c71459
Support DALI (#442) 2020-12-02 22:06:23 +08:00
QingshuChen 832364e191
support static graph train for kunlun (#441) 2020-12-02 18:36:51 +08:00
littletomatodonkey e83e3038e1
fix local rank get word size in dist (#402)
* fix local rank
* fix export model
2020-11-18 13:59:34 +08:00
littletomatodonkey 6a5f4626d7
add static running in dygraph (#399)
* add static running in dygraph
2020-11-18 09:48:56 +08:00