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

Method get_logprobs

model/diffusion/diffusion_vpg.py:343–420  ·  view source on GitHub ↗

Calculating the logprobs of the entire chain of denoised actions. Args: cond: 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) get_ent: f

(
        self,
        cond,
        chains,
        get_ent: bool = False,
        use_base_policy: bool = False,
    )

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 2

lossMethod · 0.95
lossMethod · 0.45

Calls 1

p_mean_varMethod · 0.95

Tested by

no test coverage detected