Method__init__(self, width, depth, mlp_dim, num_heads, num_kv_heads, head_dim)
lerobot_policy_mint/src/lerobot_policy_mint/modeling_mint.py:300
Method__init__(
self, codebook_size, Cvae, using_znorm, beta: float = 0.25,
default_qresi_counts=0, patch_nu
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:21
Method__init__(self, *, in_dims, out_channels=None, dropout, groups=3)
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:422
Method__init__(
self, *, ch=32, ch_mult=(1, 2, 4, 8), num_res_blocks=2,
dropout=0.0, in_dims=3, z_channels,
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:467
Method__init__(
self, *, ch=128, ch_mult=(1, 2, 4, 8), num_res_blocks=2,
dropout=0.0, in_dims=3, z_channels,
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:549
Method__init__(
self,
seq_dim,
codebook_size=512,
codebook_dim=16,
ch=32,
pa
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:633