ColossalAI/colossalai/nn/parallel
Frank Lee 15aab1476e
[zero] avoid zero hook spam by changing log to debug level (#1137)
2022-06-21 10:44:01 +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 [zero] avoid zero hook spam by changing log to debug level (#1137) 2022-06-21 10:44:01 +08:00