FoolPlayer ab8a47f830 [shardformer] add Dropout layer support different dropout pattern (#3856) ... * 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		2023-07-04 16:05:01 +08:00
..
__init__.py	[shardformer] add Dropout layer support different dropout pattern (#3856)	2023-07-04 16:05:01 +08:00
_operation.py	[shardformer] add Dropout layer support different dropout pattern (#3856)	2023-07-04 16:05:01 +08:00
dist_crossentropy.py	[shardformer] add Dropout layer support different dropout pattern (#3856)	2023-07-04 16:05:01 +08:00
dropout.py	[shardformer] add Dropout layer support different dropout pattern (#3856)	2023-07-04 16:05:01 +08:00
layers.py	[shardformer] add Dropout layer support different dropout pattern (#3856)	2023-07-04 16:05:01 +08:00