github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-12-19 02:22:47 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
c4fe9e812ed6880b51bd3a7b6dc12cf81b402a21
ColossalAI/examples/language/bert/data.py
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

4.2 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink