You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/pipeline
アマデウス 530283dba0
fix object_to_tensor usage when torch>=2.3.0 (#5820)
4 months ago
..
schedule [Feature] optimize PP overlap (#5735) 5 months ago
__init__.py
p2p.py fix object_to_tensor usage when torch>=2.3.0 (#5820) 4 months ago
stage_manager.py [Feature] optimize PP overlap (#5735) 5 months ago