mirror of https://github.com/hpcaitech/ColossalAI
ab8a47f830
* add dropout layer, add dropout test * modify seed manager as context manager * add a copy of col_nn.layer * add dist_crossentropy loss; separate module test * polish the code * fix dist crossentropy loss |
||
---|---|---|
.. | ||
config.py | ||
module_test.py | ||
test.py |