ColossalAI/colossalai/auto_parallel
YuliangLiu0306 4b29112ab2
[autoparallel] gpt2 autoparallel examples (#2267)
* [autoparallel] gpt2 autoparallel examples

* polish code

* polish code
2023-01-03 14:23:33 +08:00
..
checkpoint [autoparallel] user-friendly API for CheckpointSolver. (#1879) 2022-11-10 20:59:28 +08:00
meta_profiler [autoparallel] patch torch.flatten metainfo for autoparallel (#2247) 2023-01-02 15:51:03 +08:00
passes [autoparallel] fix construct meta info. (#2245) 2022-12-30 19:56:44 +08:00
pipeline_shard [autoparallel] init new folder structure (#1696) 2022-10-13 14:18:55 +08:00
tensor_shard [autoparallel] gpt2 autoparallel examples (#2267) 2023-01-03 14:23:33 +08:00
__init__.py [autoparallel] standardize the code structure (#1469) 2022-08-19 15:51:54 +08:00