MCPcopy Create free account

hub / github.com/ReinFlow/ReinFlow / functions

Functions864 in github.com/ReinFlow/ReinFlow

⨍ Functions 864◇ Types & classes 170↳ Endpoints 1
Method__init__
( self, actor, critic, ft_denoising_steps, ft_denoising_steps_d=0,
model/diffusion/diffusion_vpg.py:51
Method__init__
( self, network, horizon_steps, obs_dim, action_dim, network_p
model/diffusion/diffusion.py:51
Method__init__
(self)
model/diffusion/sde_lib.py:149
Method__init__
Construct a Variance Preserving SDE. Args: beta_min: value of beta(0) beta_max: value of beta(1) N: number of d
model/diffusion/sde_lib.py:181
Method__init__
( self, action_dim, base_eta=0.5, min_eta=0.1, max_eta=1.0, **
model/diffusion/eta.py:67
Method__init__
( self, input_dim, mlp_dims, activation_type="ReLU", out_activation_ty
model/diffusion/eta.py:100
Method__init__
( self, input_dim, mlp_dims, action_dim, activation_type="ReLU",
model/diffusion/eta.py:146
Method__init__
( self, gamma_denoising: float, clip_ploss_coef: float, clip_ploss_coef_base:
model/diffusion/diffusion_ppo.py:47
Method__init__
( self, actor, critic, use_ddim=False, # modifying denoising schedule
model/diffusion/diffusion_dql.py:41
Method__init__
(self, dim)
model/diffusion/modules.py:38
Method__init__
(self, dim)
model/diffusion/modules.py:54
Method__init__
( self, inp_channels, out_channels, kernel_size, n_groups=None,
model/diffusion/modules.py:77
Method__init__
( self, actor, critic, **kwargs, )
model/diffusion/diffusion_awr.py:38
Method__init__
( self, sde, clip_ploss_coef, clip_vloss_coef=None, norm_adv=True,
model/diffusion/diffusion_ppo_exact.py:45
Method__init__
( self, actor, critic, **kwargs, )
model/diffusion/diffusion_qsm.py:41
Method__init__
( self, actor, critic_q, critic_v, **kwargs, )
model/diffusion/diffusion_idql.py:46
Method__init__
( self, actor, critic, use_ddim=False, # modifying denoising schedule
model/diffusion/diffusion_dipo.py:40
Method__init__
( self, backbone:VitEncoder, # VitEncoder instance action_dim, horizon_step
model/flow/mlp_shortcut.py:221
Method__init__
( self, policy:ShortCutFlowMLP, denoising_steps:int, learn_explore_noise_from:
model/flow/mlp_shortcut.py:486
Method__init__
( self, policy:ShortCutFlowViT, denoising_steps:int, learn_explore_noise_from:
model/flow/mlp_shortcut.py:599
Method__init__
Initialize the ReFlow model with specified parameters. Args: network: FlowMLP network for velocity prediction. device
model/flow/reflow.py:43
Method__init__
(self, in_dim:int, out_dim:int, logprob_denoising_std_range
model/flow/mlp_flow.py:171
Method__init__
( self, policy:FlowMLP, denoising_steps, learn_explore_noise_from, ini
model/flow/mlp_flow.py:218
Method__init__
( self, backbone: VitEncoder, action_dim, horizon_steps, cond_dim,
model/flow/mlp_flow.py:368
Method__init__
( self, policy:VisionFlowMLP, denoising_steps, learn_explore_n
model/flow/mlp_flow.py:557
Method__init__
( self, network: ShortCutFlowMLP, device: torch.device, horizon_steps: int,
model/flow/shortcutflow.py:64
Method__init__
Initialize the OneStepActor with a neural network to map observations and noise to actions. Args: obs_dim (int): Dimension of the
model/flow/ft_baselines/fql.py:60
Method__init__
(self, device, policy, critic, actor_poli
model/flow/ft_ppo/ppoflow.py:39
Method__init__
(self, device, policy, critic, actor_poli
model/flow/ft_ppo/pposhortcut.py:36
Method__init__
( self, clip_ploss_coef: float, clip_vloss_coef: Optional[float] = None, norm_
model/gaussian/gaussian_ppo.py:43
Method__init__
( self, actor, critic, n_critics, soft_action_sample=False, so
model/gaussian/gaussian_ibrl.py:39
Method__init__
( self, clip_ploss_coef: float, clip_vloss_coef: Optional[float] = None, norm_
model/gaussian/gaussian_grpo.py:43
Method__init__
( self, actor, critic, **kwargs, )
model/gaussian/gaussian_awr.py:37
Method__init__
( self, actor, critic, n_critics, backup_entropy=False, **kwar
model/gaussian/gaussian_rlpd.py:41
Method__init__
( self, actor, **kwargs, )
model/gaussian/gaussian_vpg_grpo.py:33
Method__init__
( self, actor, critic, **kwargs, )
model/gaussian/gmm_vpg.py:29
Method__init__
( self, actor, critic, network_path=None, cql_clip_diff_min=-np.inf,
model/gaussian/gaussian_calql.py:41
Method__init__
( self, actor, **kwargs, )
model/gaussian/gaussian_rwr.py:38
Method__init__
( self, clip_ploss_coef: float, clip_vloss_coef: Optional[float] = None, norm_
model/gaussian/gmm_ppo.py:43
Method__init__
( self, actor, critic:CriticObsAct, **kwargs, )
model/gaussian/gaussian_sac.py:38
Method__init__
( self, actor, critic, **kwargs, )
model/gaussian/gaussian_vpg.py:33
Method__init__
( self, clip_ploss_coef: float, clip_vloss_coef: Optional[float] = None, norm_
model/rl/gaussian_ppo.py:43
Method__init__
( self, actor, critic, n_critics, soft_action_sample=False, so
model/rl/gaussian_ibrl.py:39
Method__init__
( self, actor, critic, **kwargs, )
model/rl/gaussian_awr.py:37
Method__init__
( self, actor, critic, n_critics, backup_entropy=False, **kwar
model/rl/gaussian_rlpd.py:41
Method__init__
( self, actor, critic, **kwargs, )
model/rl/gmm_vpg.py:29
Method__init__
( self, actor, critic, network_path=None, cql_clip_diff_min=-np.inf,
model/rl/gaussian_calql.py:41
Method__init__
( self, actor, **kwargs, )
model/rl/gaussian_rwr.py:38
Method__init__
( self, clip_ploss_coef: float, clip_vloss_coef: Optional[float] = None, norm_
model/rl/gmm_ppo.py:43
Method__init__
( self, actor, critic, **kwargs, )
model/rl/gaussian_sac.py:39
Method__init__
( self, actor, critic, **kwargs, )
model/rl/gaussian_vpg.py:36
Method__init__
( self, action_dim, horizon_steps, cond_dim=None, mlp_dims=[256, 256,
model/common/mlp_gmm.py:35
Method__init__
( self, obs_shape: List[int], cfg: VitEncoderConfig, num_channel=3, im
model/common/vit.py:50
Method__init__
(self, embed_dim, num_channel=3, img_h=96, img_w=96)
model/common/vit.py:87
Method__init__
(self, embed_dim, use_norm, num_channel=3, img_h=96, img_w=96)
model/common/vit.py:101
Method__init__
(self, embed_dim, num_head)
model/common/vit.py:125
Method__init__
(self, embed_dim, num_head, dropout)
model/common/vit.py:151
Method__init__
( self, network, horizon_steps, network_path=None, device="cuda:0",
model/common/gaussian.py:37
Method__init__
( self, backbone, action_dim, horizon_steps, cond_dim, img_con
model/common/mlp_gaussian.py:40
Method__init__
(self, num_patch, patch_dim, prop_dim, proj_dim, dropout)
model/common/modules.py:33
Method__init__
(self, pad)
model/common/modules.py:67
Method__init__
( self, cond_dim, mlp_dims, action_dim, action_steps=1, activa
model/common/critic.py:84
Method__init__
( self, backbone, cond_dim, img_cond_steps=1, spatial_emb=128,
model/common/critic.py:143
Method__init__
( self, dim_list, activation_type="Mish", out_activation_type="Identity",
model/common/mlp.py:121
Method__init__
( self, hidden_dim, activation_type="Mish", use_layernorm=False, dropo
model/common/mlp.py:166
Method__init__
( self, action_dim, horizon_steps, cond_dim, transformer_embed_dim=256
model/common/transformer.py:39
Method__init__
( self, action_dim, horizon_steps, cond_dim, num_modes=5, tran
model/common/transformer.py:125
Method__init__
( self, network, horizon_steps, network_path=None, device="cuda:0",
model/common/gmm.py:37
Method__init__
(self, env_type, **kwargs)
env/plot_traj.py:36
Method__init__
( self, env_fns, dummy_env_fn=None, observation_space=None, action_spa
env/gym_utils/async_vector_env.py:151
Method__init__
(self, num_envs, observation_space, action_space)
env/gym_utils/vector_env.py:58
Method__init__
(self, env)
env/gym_utils/vector_env.py:263
Method__init__
(self, env_fns, observation_space=None, action_space=None, copy=True)
env/gym_utils/sync_vector_env.py:78
Method__init__
(self)
env/gym_utils/furniture_normalizer.py:34
Method__init__
( self, env, normalization_path, )
env/gym_utils/wrapper/mujoco_locomotion_lowdim.py:35
Method__init__
( self, env: FurnitureRLSimEnv, n_obs_steps=1, n_action_steps=1, max_e
env/gym_utils/wrapper/furniture.py:72
Method__init__
( self, env, normalization_path, )
env/gym_utils/wrapper/d3il_lowdim.py:37
Method__init__
( self, env, normalization_path=None, low_dim_keys=[ "robot0_eef_p
env/gym_utils/wrapper/robomimic_lowdim.py:40
Method__init__
( self, env, n_obs_steps=1, n_action_steps=1, max_episode_steps=None,
env/gym_utils/wrapper/multi_step.py:108
Method__init__
( self, env, shape_meta: dict, normalization_path=None, low_dim_keys=[
env/gym_utils/wrapper/robomimic_image.py:39
Method__init__
(self, log_file)
data_process/hdf5_to_npz.py:39
Method__init__
(self, epsilon=1e-4, shape=(), device=None)
util/reward_scaling_ts.py:30
Method__init__
(self, num_envs, cliprew=10.0, gamma=0.99, epsilon=1e-8, per_env=False, device=None)
util/reward_scaling_ts.py:56
Method__init__
( self, optimizer: torch.optim.Optimizer, first_cycle_steps: int, cycle_mult:
util/scheduler.py:44
Method__init__
( self, epsilon=1e-4, # initial count (with mean=0 ,var=1) shape=(), # unbatched sha
util/reward_scaling.py:37
Method__init__
(self, num_envs, cliprew=10.0, gamma=0.99, epsilon=1e-8, per_env=False)
util/reward_scaling.py:74
Method__init__
(self)
util/timer.py:51
Method__init__
( self, optimizer: torch.optim.Optimizer, schedule_type: str, **kwargs )
util/scheduler_simple.py:29
Method__init__
( self, dataset_path, max_n_episodes=10000, discount_factor=1.0, devic
agent/dataset/sequence.py:192
Method__init__
( self, data_directory: os.PathLike, # data='train', device="cpu", obs
agent/dataset/d3il_dataset/stacking_dataset.py:224
Method__init__
( self, data_directory: os.PathLike, device="cpu", obs_dim: int = 20,
agent/dataset/d3il_dataset/base_dataset.py:16
Method__init__
( self, data_directory: os.PathLike, device="cpu", obs_dim: int = 20,
agent/dataset/d3il_dataset/pushing_dataset.py:14
Method__init__
( self, data_directory: os.PathLike, device="cpu", obs_dim: int = 20,
agent/dataset/d3il_dataset/sorting_dataset.py:288
Method__init__
( self, data_directory: os.PathLike, device="cpu", obs_dim: int = 20,
agent/dataset/d3il_dataset/avoiding_dataset.py:12
Method__init__
( self, data_directory: os.PathLike, device="cpu", obs_dim: int = 20,
agent/dataset/d3il_dataset/aligning_dataset.py:170
Method__init__
(self, cfg)
agent/eval/eval_diffusion_agent.py:55
Method__init__
(self, cfg)
agent/eval/eval_diffusion_img_agent.py:54
Method__init__
(self, cfg)
agent/eval/eval_reflow_agent.py:34
Method__init__
(self, cfg)
agent/eval/eval_reflow_img_agent.py:34
Method__init__
(self, cfg)
agent/eval/eval_shortcut_agent.py:35
← previousnext →401–500 of 864, ranked by callers