ColossalAI/colossalai/pipeline
アマデウス 530283dba0 fix object_to_tensor usage when torch>=2.3.0 (#5820) 2024-07-16 13:59:25 +08:00
..
schedule [Feature] optimize PP overlap (#5735) 2024-06-26 14:48:02 +08:00
__init__.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
p2p.py fix object_to_tensor usage when torch>=2.3.0 (#5820) 2024-07-16 13:59:25 +08:00
stage_manager.py [Feature] optimize PP overlap (#5735) 2024-06-26 14:48:02 +08:00