Commit Graph

3 Commits

Author SHA1 Message Date
hiyouga
2ed8270112 clean code 2024-06-13 01:58:16 +08:00
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
308edbc426 rename package 2024-05-16 18:39:08 +08:00