github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-12 23:22:42 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
86b63f720cf60deefe40874517b3d8e1dccb7af3
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