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

Method __init__

model/diffusion/diffusion.py:51–113  ·  view source on GitHub ↗
(
        self,
        network,
        horizon_steps,
        obs_dim,
        action_dim,
        network_path=None,
        device="cuda:0",
        # Various clipping
        denoised_clip_value=1.0,
        randn_clip_value=10,
        final_action_clip_value=None,
        eps_clip_value=None,  # DDIM only
        # DDPM parameters
        denoising_steps=100,
        predict_epsilon=True,
        # DDIM sampling
        use_ddim=False,
        ddim_discretize="uniform",
        ddim_steps=None,
        **kwargs,
    )

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 3

calculate_parametersMethod · 0.95
load_state_dictMethod · 0.80
loadMethod · 0.45

Tested by

no test coverage detected