Oneflow-Inc / libai

LiBai(李白): A Toolbox for Large-Scale Distributed Parallel Training
https://libai.readthedocs.io
Apache License 2.0
391 stars 55 forks source link

add check_meta and sync_data for to_global #530

Open hanwen-sun opened 10 months ago

hanwen-sun commented 10 months ago

对照oneflow的这个pr: https://github.com/Oneflow-Inc/oneflow/pull/10430 在libai的 get_batch中的to_global添加对应的check_meta与sync_data参数