trinity.trainer.verl package

Submodules

• trinity.trainer.verl.dp_actor module
  • DataParallelPPOActor
    • DataParallelPPOActor.__init__()
    • DataParallelPPOActor.set_algorithm()
    • DataParallelPPOActor.update_policy()
    • DataParallelPPOActor.compute_log_prob()
• trinity.trainer.verl.fsdp_checkpoint_manager module
  • FSDPCheckpointManager
    • FSDPCheckpointManager.__init__()
    • FSDPCheckpointManager.upload_state_dict()
    • FSDPCheckpointManager.save_state_dict()
    • FSDPCheckpointManager.save_checkpoint()
    • FSDPCheckpointManager.wait_on_save_thread()
• trinity.trainer.verl.fsdp_workers module
• trinity.trainer.verl.megatron_actor module
• trinity.trainer.verl.megatron_checkpoint_manager module
• trinity.trainer.verl.megatron_workers module
• trinity.trainer.verl.utils module
  • to_data_proto()
  • compute_data_metrics()
  • get_latest_hf_checkpoint_path()
  • hf_processor()
  • apply_fsdp2()

Module contents

trinity.trainer.verl.patch_for_transformers_v5()