mirrors / dmlc--dgl

项目文件夹

dmlc--dgl
文件
0.6.x
T
文件历史
Lingfan Yu 29dd22e666 [Model] Support Multi-GPU for Transformer model (#356) 
* multi-process version of transformer

* lots of fix

* fix bugs and accum gradients for multiple batches

* many fixes

* minor

* upd

* set torch device

* fix bugs

* fix and minor

* comments and clean up

* uncomment viz code
2019-02-11 20:17:45 -05:00
..
__init__.py
[Model] Support Multi-GPU for Transformer model (#356)
2019-02-11 20:17:45 -05:00