github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-16 15:37:57 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
6e38eafebec514cd670758a0fb06b37bd01d224e
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