hiyouga 4f3c89a6eb fix ppo trainer save zero3 model ...

accelerator.get_state_dict(ds_model) should be called at all ranks


Former-commit-id: 4489d73ac75c8dbc002fc16c854148994d432c3a

2024-06-07 05:14:19 +08:00

..

__init__.py

rename package

2024-05-16 18:39:08 +08:00

metric.py

fix ppo trainer save zero3 model

2024-06-07 05:14:19 +08:00

trainer.py

rename files

2024-06-07 00:09:06 +08:00

workflow.py

rename files

2024-06-07 00:09:06 +08:00