ColossalAI/colossalai/nn/parallel
ver217 895c1c5ee7
[tensor] refactor param op hook (#1097)
* refactor param op hook

* add docstr

* fix bug
2022-06-13 16:11:53 +08:00
..
layers [refactory] add nn.parallel module (#1068) 2022-06-06 15:34:41 +08:00
__init__.py [refactory] add nn.parallel module (#1068) 2022-06-06 15:34:41 +08:00
data_parallel.py [tensor] refactor param op hook (#1097) 2022-06-13 16:11:53 +08:00