MCPcopy Create free account
hub / github.com/ReinFlow/ReinFlow / loss

Method loss

model/diffusion/diffusion_ppo.py:79–221  ·  view source on GitHub ↗

PPO loss obs: dict with key state/rgb; more recent obs at the end state: (B, To, Do) rgb: (B, To, C, H, W) chains: (B, K+1, Ta, Da) returns: (B, ) values: (B, ) advantages: (B,) oldlogprobs: (B, K, Ta, Da) use_

(
        self,
        obs,
        chains_prev,
        chains_next,
        denoising_inds,
        returns,
        oldvalues,
        advantages,
        oldlogprobs,
        use_bc_loss=False,
        reward_horizon=4,
    )

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 3

forwardMethod · 0.45
get_logprobsMethod · 0.45

Tested by

no test coverage detected