Functions — github.com/GeWu-Lab/AnyTouch2

CodeHub MCPcopy Create free account

hub / github.com/GeWu-Lab/AnyTouch2 / functions

Functions813 in github.com/GeWu-Lab/AnyTouch2

⨍ Functions 813◇ Types & classes 138↳ Endpoints 2

( self, out_dim, student_temp=0.1, center_momentum=0.9, )

sparsh/tactile_ssl/loss/dino_loss.py:12

(self, window_size=20, fmt=None)

util/misc.py:29

(self, delimiter="\t")

util/misc.py:87

util/misc.py:254

(self, dataset, num_replicas=None, rank=None, shuffle=True, drop_last: bool = False)

util/sampler.py:7

util/sampler.py:11

data/downstream_dataset.py:90

data/downstream_dataset.py:194

data/downstream_dataset.py:317

sparsh/tactile_ssl/data/vision_based_grasp_probes.py:57

sparsh/tactile_ssl/data/vision_based_textile_probes.py:59

sparsh/tactile_ssl/data/vision_based_pose_probes.py:141

util/misc.py:77

util/misc.py:108

Function_get_conv1d_output_size

(input_size, kernel_size, stride, padding, dilation)

sparsh/tactile_ssl/utils/__init__.py:191

Method_get_trajectory_pose_force

(self, idx_trajectory)

sparsh/tactile_ssl/data/vision_based_forces_slip_probes.py:264

Method_getitem_trajectory

(self, idx_trajectory, idx_sample)

sparsh/tactile_ssl/data/vision_based_slip_probes.py:189

Method_getitem_trajectory

(self, idx_trajectory, idx_sample)

sparsh/tactile_ssl/data/vision_based_forces_slip_probes.py:247

Method_init_weights

model/linear_probe.py:79

Method_init_weights

sparsh/tactile_ssl/model/layers/dino_head.py:37

Method_init_weights

sparsh/tactile_ssl/downstream_task/attentive_pooler.py:85

Method_len_trajectory

(self, idx_trajectory)

sparsh/tactile_ssl/data/vision_based_slip_probes.py:121

Method_len_trajectory

(self, idx_trajectory)

sparsh/tactile_ssl/data/vision_based_forces_slip_probes.py:137

Method_plot_tactile_clip

sparsh/tactile_ssl/data/vision_based_interactive.py:140

Method_sigterm_handler_fn

(self, signum: _SIGNUM, _: FrameType)

sparsh/tactile_ssl/utils/signal_connector.py:125

Method_sigterm_notifier_fn

(self, signum: _SIGNUM, _: FrameType)

sparsh/tactile_ssl/utils/signal_connector.py:111

Method_slurm_sigusr_handler_fn

(self, signum: _SIGNUM, _: FrameType)

sparsh/tactile_ssl/utils/signal_connector.py:76

Functionadd_residual

( x: Tensor, brange: Tensor, residual: Tensor, residual_scale_factor: float, scaling_vecto

model/layers/decoder_block.py:179

Functionadd_residual

( x: Tensor, brange: Tensor, residual: Tensor, residual_scale_factor: float, scaling_vecto

sparsh/tactile_ssl/model/layers/decoder_block.py:182

Functionadjust_learning_rate

Decay the learning rate with half-cycle cosine after warmup

util/lr_sched.py:9

Functionall_reduce_mean

util/misc.py:332

Methodattn_residual_func

model/layers/block.py:87

Methodattn_residual_func

(x: Tensor, attn_bias=None)

model/layers/block.py:217

Methodattn_residual_func

sparsh/tactile_ssl/model/layers/block.py:90

Methodattn_residual_func

(x: Tensor, attn_bias=None)

sparsh/tactile_ssl/model/layers/block.py:220

util/misc.py:61

(input_dim, output_dim)

model/layers/gumbel_vector_quantizer.py:66

(input_dim, output_dim)

sparsh/tactile_ssl/model/layers/gumbel_vector_quantizer.py:66

Functioncheck_saved_dataset

sparsh/scripts/download_datasets_scratch/compress_feeling_success.py:55

Functioncheck_saved_dataset

sparsh/scripts/download_datasets_scratch/compress_object_folder.py:50

Functioncheck_saved_dataset

sparsh/scripts/download_datasets_scratch/compress_touch_go.py:65

model/layers/gumbel_vector_quantizer.py:116

sparsh/tactile_ssl/model/layers/gumbel_vector_quantizer.py:116

Methodconfigure_optimizers

( # pyright: ignore[reportIncompatibleMethodOverride] self, num_iterations_per_epoch, num_epochs

sparsh/tactile_ssl/algorithm/ijepa.py:381

Methodconfigure_optimizers

( # pyright: ignore[reportIncompatibleMethodOverride] self, num_iterations_per_epoch, num_epochs

sparsh/tactile_ssl/algorithm/dinov2.py:518

Methodconfigure_optimizers

( # pyright: ignore[reportIncompatibleMethodOverride] self, num_iterations_per_epoch, num_epochs

sparsh/tactile_ssl/algorithm/dino.py:440

Methodconfigure_optimizers

( self, num_iterations_per_epoch, num_epochs, )

sparsh/tactile_ssl/algorithm/vjepa.py:382

Methodconfigure_optimizers

( self, num_iterations_per_epoch, num_epochs )

sparsh/tactile_ssl/algorithm/mae.py:160

Methodconfigure_optimizers

( self, num_iterations_per_epoch, num_epochs, *args, **kwargs )

sparsh/tactile_ssl/downstream_task/sl_module.py:118

Methodconstrain_mask

Helper to restrict given mask to a set of acceptable regions

sparsh/tactile_ssl/algorithm/ijepa.py:159

Methodconstrain_mask

Helper to restrict given mask to a set of acceptable regions

sparsh/tactile_ssl/algorithm/dinov2.py:244

Methodconstrain_mask

Helper to restrict given mask to a set of acceptable regions

sparsh/tactile_ssl/algorithm/dino.py:230

Methodcount_parameters

sparsh/tactile_ssl/algorithm/vjepa.py:86

Methodcross_attn_residual_func

(q: Tensor, kv: Tensor)

model/layers/decoder_block.py:108

Methodcross_attn_residual_func

(q: Tensor, kv: Tensor)

sparsh/tactile_ssl/model/layers/decoder_block.py:111

Methoddel_decoder

sparsh/tactile_ssl/model/point_mae.py:910

Functiondraw_3d_axes

Draw 3D axes in 3D matplotlib plot. Args: ax: matplotlib 3D axis world_T_camera: n x 4 x 4 transformation matrix from camera

sparsh/tactile_ssl/utils/plotting_utils.py:50

Methodemb_forward

(self, pixel_values: Optional[torch.FloatTensor] = None, noise=None, sensor_type=None, use_mask = True)

model/tactile_mae.py:118

Methodemb_forward

(self, pixel_values: Optional[torch.FloatTensor] = None, points=None, points_num=None, noise=None, sensor_type

sparsh/tactile_ssl/model/point_mae.py:499

Methodemb_forward

(self, pixel_values: Optional[torch.FloatTensor] = None, points=None, points_num=None, noise=None, sensor_type

sparsh/tactile_ssl/model/point_mae.py:1049

Methodemb_forward

(self, pixel_values: torch.FloatTensor, noise=None, sensor_type=None)

sparsh/tactile_ssl/model/point_mae.py:1508

(*args, **kwargs)

sparsh/tactile_ssl/model/multimodal_transformer.py:90

(*args, **kwargs)

sparsh/tactile_ssl/model/vision_transformer.py:193

Methodffn_residual_func

model/layers/block.py:90

Methodffn_residual_func

(x: Tensor, attn_bias=None)

model/layers/block.py:220

Methodffn_residual_func

model/layers/decoder_block.py:111

Methodffn_residual_func

sparsh/tactile_ssl/model/layers/block.py:93

Methodffn_residual_func

(x: Tensor, attn_bias=None)

sparsh/tactile_ssl/model/layers/block.py:223

Methodffn_residual_func

sparsh/tactile_ssl/model/layers/decoder_block.py:114

sparsh/tactile_ssl/model/layers/patch_embed.py:103

(self, x=None, sensor_type=None, probe=False, get_cls=False)

model/tactile_mae.py:144

model/linear_probe.py:92

model/linear_probe.py:122

model/linear_probe.py:146

(self, x, sensor_type = None, return_feature = False)

model/linear_probe.py:215

model/layers/drop_path.py:33

(self, x_or_x_list)

model/layers/block.py:249

(self, x: Tensor)

model/layers/layer_scale.py:26

(self, x: Tensor, attn_bias=None)

model/layers/attention.py:78

(self, q: Tensor, x: Tensor)

model/layers/attention.py:120

(self, q: Tensor, x: Tensor, attn_bias=None)

model/layers/attention.py:152

model/layers/attention.py:193

(self, x: Tensor)

model/layers/mlp.py:34

model/layers/decoder_block.py:104

(self, x: Tensor)

model/layers/swiglu_ffn.py:30

(self, x: torch.Tensor)

sparsh/train_task.py:47

(self, x, masks=None)

sparsh/tactile_ssl/utils/masking.py:26

(self, ctxt, masks_ctxt, masks_tgt)

sparsh/tactile_ssl/utils/masking.py:44

(self, xs, masks=None, mask_types=None)

sparsh/tactile_ssl/model/multimodal_transformer.py:323

( self, xs: List[torch.Tensor], ids_restore: List[torch.Tensor], mask_types: L

sparsh/tactile_ssl/model/multimodal_transformer.py:439

sparsh/tactile_ssl/model/point_mae.py:61

Args: hidden_states (`torch.FloatTensor`): input to the layer of shape `(batch, seq_len, embed_dim)` attention_mask (

sparsh/tactile_ssl/model/point_mae.py:89

r""" Args: inputs_embeds (`torch.FloatTensor` of shape `(batch_size, sequence_length, hidden_size)`): Optionally,

sparsh/tactile_ssl/model/point_mae.py:189

(self, x=None, points=None, points_num=None, sensor_type=None, data_type=None, target_sensor_type = None)

sparsh/tactile_ssl/model/point_mae.py:571

sparsh/tactile_ssl/model/point_mae.py:1128

(self, x, sensor_type = None)

sparsh/tactile_ssl/model/point_mae.py:1529

sparsh/tactile_ssl/model/pretrained.py:42

sparsh/tactile_ssl/model/vision_transformer.py:50

(self, *args, **kwargs)

sparsh/tactile_ssl/model/vision_transformer.py:420

(self, x, context_masks, masks=None, mask_index=1)

sparsh/tactile_ssl/model/vision_transformer.py:448

← previous next →501–600 of 813, ranked by callers