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
..
__init__.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
_utils.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
base.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
generate.py	[inference] refactor examples and fix schedule (#5077)	2023-11-21 10:46:03 +08:00
interleaved_pp.py	[pipeline]: support arbitrary batch size in forward_only mode (#5201)	2024-01-02 23:41:12 +08:00
one_f_one_b.py	[pipeline]: support arbitrary batch size in forward_only mode (#5201)	2024-01-02 23:41:12 +08:00