Wenhao Chen 3c0d82b19b [pipeline]: support arbitrary batch size in forward_only mode (#5201) ... * fix: remove drop last in val & test dataloader * feat: add run_forward_only, support arbitrary bs * chore: modify ci script		2024-01-02 23:41:12 +08:00
..
schedule	[pipeline]: support arbitrary batch size in forward_only mode (#5201)	2024-01-02 23:41:12 +08:00
__init__.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
p2p.py	[pipeline]: fix p2p comm, add metadata cache and support llama interleaved pp (#5134)	2023-12-22 10:44:00 +08:00
stage_manager.py	[pipeline]: fix p2p comm, add metadata cache and support llama interleaved pp (#5134)	2023-12-22 10:44:00 +08:00