trinity.trainer.verl package

Submodules

• trinity.trainer.verl.checkpoint module
  • CheckpointCoordinator
    • CheckpointCoordinator.__init__()
    • CheckpointCoordinator.save_async()
    • CheckpointCoordinator.save_sync()
    • CheckpointCoordinator.wait_all()
    • CheckpointCoordinator.register_and_monitor()
• trinity.trainer.verl.config module
  • build_verl_config()
• trinity.trainer.verl.fsdp_engine module
  • fsdp_save_state_dict()
  • fsdp_upload_state_dict()
  • fsdp_sync_weight_nccl()
• trinity.trainer.verl.losses module
  • TrinityPolicyLoss
    • TrinityPolicyLoss.__init__()
  • build_trinity_loss()
• trinity.trainer.verl.megatron_engine module
  • megatron_save_state_dict()
  • megatron_upload_state_dict()
  • megatron_sync_weight_nccl()
• trinity.trainer.verl.monkey_patch module
  • save_checkpoint()
  • patch_verl_engine()
• trinity.trainer.verl.trainer module
• trinity.trainer.verl.utils module
  • to_data_proto()
  • compute_data_metrics()
• trinity.trainer.verl.workers module

Module contents

Verl v0.8.0 trainer