Commit Graph

3 Commits

Author SHA1 Message Date
hiyouga
4489d73ac7 fix ppo trainer save zero3 model
accelerator.get_state_dict(ds_model) should be called at all ranks
2024-06-07 05:14:19 +08:00
hiyouga
74f96efef9 rename files 2024-06-07 00:09:06 +08:00
hiyouga
308edbc426 rename package 2024-05-16 18:39:08 +08:00