trinity.trainer.verl.monkey_patch module

trinity.trainer.verl.monkey_patch.save_checkpoint(self, local_path: str, hdfs_path: str | None = None, global_step: int = 0, max_ckpt_to_keep: int | None = None, **kwargs) → None

Save FSDP checkpoint, handling parameter offload as needed.

trinity.trainer.verl.monkey_patch.patch_verl_engine(engine)