MCPcopy Create free account

hub / github.com/RenMing-Huang/MINT / functions

Functions488 in github.com/RenMing-Huang/MINT

⨍ Functions 488◇ Types & classes 53
Method__init__
MinMax normalizer that scales actions to [-1, 1]. Args: action_dim: Action dimension min_v: Absolute minimum values (fallback if quantil
SDAT/utils/normalizer.py:129
Method__init__
(self, action_dim: int)
SDAT/utils/normalizer.py:205
Method__init__
( self, mixed_precision: int, optimizer: torch.optim.Optimizer, names: List[
SDAT/optim/amp_opt.py:26
Method__init__
( self, codebook_size, Cvae, znorm, beta: float = 0.25, default_qresi_counts=0, patch_nums=Non
SDAT/model/quant.py:12
Method__init__
(self, qresi: Phi)
SDAT/model/quant.py:199
Method__init__
(self, qresi_ls: nn.ModuleList)
SDAT/model/quant.py:208
Method__init__
(self, qresi: List)
SDAT/model/quant.py:222
Method__init__
(self, d_embed: int = 64, dropout: float = 0.0, norm_type: str = 'layer')
SDAT/model/basic_vae.py:16
Method__init__
(self, d_embed: int = 64, dropout: float = 0.0)
SDAT/model/basic_vae.py:79
Method__init__
(self, in_dims)
SDAT/model/basic_vae.py:135
Method__init__
(self, in_dims)
SDAT/model/basic_vae.py:143
Method__init__
(self, *, in_dims, out_channels=None, dropout)
SDAT/model/basic_vae.py:151
Method__init__
(self, *, in_dims, out_channels=None, dropout, groups=3)
SDAT/model/basic_vae.py:174
Method__init__
(self, in_dims)
SDAT/model/basic_vae.py:197
Method__init__
( self, *, ch=128, ch_mult=(1, 2, 4, 8), num_res_blocks=2, dropout=0.0, in_dims=3, z_channels,
SDAT/model/basic_vae.py:301
Method__init__
( self, seq_dim, codebook_size=4096, z_channels=32, ch=128, dr
SDAT/model/vqvae.py:8
Method__post_init__
(self)
lerobot_policy_mint/src/lerobot_policy_mint/configuration_mint.py:72
Method__str__
(self)
SDAT/utils/misc.py:247
Method__str__
(self)
SDAT/utils/misc.py:281
Method_init_weights
(self, m)
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:288
Method_init_weights
(self, m)
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:353
Method_init_weights
(self, m)
SDAT/model/basic_vae.py:40
Method_init_weights
(self, m)
SDAT/model/basic_vae.py:105
Methodaction_delta_indices
(self)
lerobot_policy_mint/src/lerobot_policy_mint/configuration_mint.py:202
FunctionaddClasses
(element, classes)
docs/static/js/bulma-carousel.js:95
Methodadd_chunk
(self, chunk: torch.Tensor, intention_vector: torch.Tensor)
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:867
Functionall_gather_object
Gathers picklable objects from all processes and stores them in object_list. Args: object_list (list): A list to store the gathered
SDAT/utils/dist.py:214
Functionallgather
(t: torch.Tensor, cat=True)
SDAT/utils/dist.py:110
Functionallgather_diff_shape
(t: torch.Tensor, cat=True)
SDAT/utils/dist.py:123
FunctionattachEndHoldRestart
(videoEl, iconEl, finalState)
docs/static/js/index.js:682
Functionauto_resume
(args: Any, pattern='ckpt*.pth')
SDAT/utils/misc.py:357
Methodautoregressive_sample
( self, prefix_pad_masks, past_key_values, sample_top_k: int | None = None,
lerobot_policy_mint/src/lerobot_policy_mint/modeling_mint.py:816
Methodavg
(self)
SDAT/utils/misc.py:228
Functionbl
(c)
docs/static/js/fontawesome.all.min.js:5
Functionbroadcast
(t: torch.Tensor, src_rank)
SDAT/utils/dist.py:150
FunctionbulmaCarousel
(selector)
docs/static/js/bulma-carousel.js:504
FunctionbulmaSlider
(selector)
docs/static/js/bulma-slider.js:105
Functioncamelize
(str)
docs/static/js/bulma-carousel.js:1008
Methodclose
(self)
SDAT/utils/misc.py:176
Functioncompute_action_cdf_knots_from_loader
Estimate per-dimension empirical CDF with fixed quantile knots. Uses reservoir sampling up to max_samples for memory efficiency. Returns dict with
SDAT/utils/normalizer.py:392
Functioncompute_action_stats_from_loader
Compute per-dimension mean, std, min, max, and quantiles over actions in a DataLoader. Assumes actions are under key 'actions' in each batch and th
SDAT/utils/normalizer.py:290
Methodcompute_final_norms
(inputs_embeds, adarms_cond)
lerobot_policy_mint/src/lerobot_policy_mint/modeling_mint.py:507
FunctioncopyBibTeX
()
docs/static/js/index.js:112
Functioncreate_npz_from_sample_folder
Builds a single .npz file from a folder of .png samples. Refer to DiT.
SDAT/utils/misc.py:373
Functioncreate_sinusoidal_pos_embedding
Computes sine-cosine positional embedding vectors for scalar positions.
lerobot_policy_mint/src/lerobot_policy_mint/modeling_mint.py:79
Functioncss
(element, obj)
docs/static/js/bulma-carousel.js:169
Functiondefer
()
docs/static/js/bulma-carousel.js:995
FunctiondetectSupportsPassive
()
docs/static/js/bulma-carousel.js:189
Functiondist_fmt_vals
(val: float, fmt: Union[str, None] = '%.2f')
SDAT/utils/dist.py:160
Methoddo_nothing
(*args, **kwargs)
SDAT/utils/misc.py:119
Functionec
(c)
docs/static/js/fontawesome.all.min.js:5
Functionecho
(info)
SDAT/utils/misc.py:19
Methodeini
(self, eini)
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:50
Methodeini
(self, eini)
SDAT/model/quant.py:41
Methodembed_to_fhat
(self, ms_h_BCH: List[torch.Tensor], all_to_max_scale=True, last_one=False)
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:124
Methodeval_ep
Run a lightweight eval over up to max_batches from ld_val. Returns averaged metrics: loss_rec_l1, loss_rec_l2, loss_vq, loss_vae.
SDAT/trainer.py:53
Methodextra_repr
(self)
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:54
Methodextra_repr
(self)
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:240
Methodextra_repr
(self)
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:253
Methodextra_repr
(self)
SDAT/model/quant.py:45
Methodextra_repr
(self)
SDAT/model/quant.py:217
Methodextra_repr
(self)
SDAT/model/quant.py:230
Functionfinalize
()
SDAT/utils/dist.py:210
Functionfl
(c)
docs/static/js/fontawesome.all.min.js:5
Methodflush
(self)
SDAT/utils/misc.py:173
Functionfor_visualize
(func)
SDAT/utils/dist.py:198
Methodforward
( self, attention_mask: torch.Tensor | None = None, position_ids: torch.LongTensor | N
lerobot_policy_mint/src/lerobot_policy_mint/modeling_mint.py:434
Methodforward
Do a full training forward pass and compute the loss.
lerobot_policy_mint/src/lerobot_policy_mint/modeling_mint.py:701
Methodforward
( self, f_BCH: torch.Tensor, ret_usages: bool = False, ret_fhat_scales: bool =
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:58
Methodforward
(self, h_BCH)
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:217
Methodforward
x: (B, T, 7)
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:299
Methodforward
z: (B, 3*D, T)
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:359
Methodforward
(self, x)
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:387
Methodforward
(self, x)
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:395
Methodforward
(self, x)
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:415
Methodforward
(self, x)
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:439
Methodforward
(self, x)
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:453
Methodforward
(self, x)
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:528
Methodforward
(self, z)
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:614
Methodforward
(self, inp, ret_usages: bool = False, ret_ms_l1: bool = False)
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:705
Methodforward
( self, f_BCH: torch.Tensor, ret_usages: bool = False, ret_fhat_scales: bool =
SDAT/model/quant.py:49
Methodforward
(self, h_BCH)
SDAT/model/quant.py:194
Methodforward
x: (B, T, 7)
SDAT/model/basic_vae.py:51
Methodforward
z: (B, 3*D, T)
SDAT/model/basic_vae.py:111
Methodforward
(self, x)
SDAT/model/basic_vae.py:139
Methodforward
(self, x)
SDAT/model/basic_vae.py:147
Methodforward
(self, x)
SDAT/model/basic_vae.py:167
Methodforward
(self, x)
SDAT/model/basic_vae.py:191
Methodforward
(self, x)
SDAT/model/basic_vae.py:205
Methodforward
(self, x)
SDAT/model/basic_vae.py:280
Methodforward
(self, z)
SDAT/model/basic_vae.py:366
Methodforward
(self, inp, ret_usages: bool = False, ret_ms_l1: bool = False)
SDAT/model/vqvae.py:85
Methodforward_func
(prefix_embs, suffix_embs, att_2d_masks_4d, position_ids)
lerobot_policy_mint/src/lerobot_policy_mint/modeling_mint.py:722
Methodfrom_pretrained
Override the from_pretrained method to handle key remapping and display important disclaimer.
lerobot_policy_mint/src/lerobot_policy_mint/modeling_mint.py:1037
FunctiongetNodeIndex
(node)
docs/static/js/bulma-carousel.js:1005
Functionget_device
()
SDAT/utils/dist.py:65
Methodget_ensembled_actions
(self)
lerobot_policy_mint/src/lerobot_policy_mint/mint_utils.py:876
Methodget_next_autoregressive_input
(self, si: int, SN: int, f_hat: torch.Tensor, h_BCH: torch.Tensor)
SDAT/model/quant.py:158
Methodget_optim_params
(self)
lerobot_policy_mint/src/lerobot_policy_mint/modeling_mint.py:1207
Methodget_optimizer_preset
(self)
lerobot_policy_mint/src/lerobot_policy_mint/configuration_mint.py:180
← previousnext →301–400 of 488, ranked by callers