ColossalAI/colossalai/shardformer/test
FoolPlayer ef1537759c [shardformer] add gpt2 policy and modify shard and slicer to support (#3883)
* add gpt2 policy and modify shard and slicer to support

* remove unused code

* polish code
2023-06-08 15:01:34 +08:00
..
config.py [shardformer]: Feature/shardformer, add some docstring and readme (#3816) 2023-06-08 15:01:34 +08:00
module_test.py [shardformer] add Dropout layer support different dropout pattern (#3856) 2023-06-08 15:01:34 +08:00
test.py [shardformer] add gpt2 policy and modify shard and slicer to support (#3883) 2023-06-08 15:01:34 +08:00