Functions — github.com/Abraham190137/TactileACT

hub / github.com/Abraham190137/TactileACT / functions

Functions267 in github.com/Abraham190137/TactileACT

Replace all submodules selected by the predicate with the output of func. predicate: Return true if the module is to be replaced. fu

clip_pretraining.py:13

↓ 1 callersFunctionreplace_submodules

Replace all submodules selected by the predicate with the output of func. predicate: Return true if the module is to be replaced. fu

clip_pretraining_no_pos.py:7

↓ 1 callersFunctionreplace_submodules

Replace all submodules selected by the predicate with the output of func. predicate: Return true if the module is to be replaced. fu

failed_DDP.py:18

↓ 1 callersFunctionreplace_submodules

Replace all submodules selected by the predicate with the output of func. predicate: Return true if the module is to be replaced. fu

diffusion/network.py:296

↓ 1 callersFunctionreplace_submodules

Replace all submodules selected by the predicate with the output of func. predicate: Return true if the module is to be replaced. fu

diffusion/clip_pretraining.py:21

↓ 1 callersMethodreset_parameters

(self)

detr/detr/models/position_encoding.py:65

↓ 1 callersFunctionrun_clip_pretraining

()

clip_pretraining.py:599

↓ 1 callersFunctionrun_clip_pretraining

()

clip_pretraining_no_pos.py:595

↓ 1 callersFunctionsave_norm_stats

(save_folder, num_episodes = None)

process_data_cage.py:156

↓ 1 callersFunctionsave_norm_stats

(save_folder)

diffusion/process_data_cage.py:240

↓ 1 callersFunctionset_seed

(seed)

utils.py:292

↓ 1 callersFunctionsetup_for_distributed

This function disables printing when not in master process

detr/detr/util/misc.py:363

↓ 1 callersFunctiontrain

(num_epochs,camera_names,nets:nn.ModuleDict,train_dataloader,val_dataloader:torch.utils.data.dataloader,enc_ty

diffusion/train_policy.py:216

↓ 1 callersFunctiontrain_bc

(policy:ACTPolicy, train_dataloader: DataLoader, val_dataloader: DataLoader,

imitate_episodes.py:214

↓ 1 callersFunctionvisualize

(images, qpos, actions, ground_truth=None)

robot_operation.py:141

↓ 1 callersFunctionvisualize

(images, qpos, actions, ground_truth=None)

diffusion/robot_operation_old.py:95

↓ 1 callersFunctionvisualize_data

(image_data, qpos_data, action_pred_data, is_pad_data, action_gt_data=None)

visualization_utils.py:41

(self, qpos:torch.Tensor, images, actions=None, is_pad=None, z=None, ignore_latent=False)

policy.py:140

Method__call__

(self, qpos, action)

diffusion/utils.py:21

Method__call__

(self, qpos, action)

diffusion/utils.py:38

Method__call__

(self, qpos, action)

diffusion/dataset.py:15

Method__getattr__

(self, attr)

detr/detr/util/misc.py:171

Method__getitem__

(self, index)

clip_pretraining.py:227

Method__getitem__

(self, index)

clip_pretraining_no_pos.py:209

diffusion/clip_pretraining.py:157

Method__getitem__

(self, index)

diffusion/dataset.py:63

Method__init__

(self, backbones, transformer, encoder, state_dim, num_queries, camera_names, z_dimension, cam_backbone_mappin

detrvae.py:30

Method__init__

(self, print=False, plot=False, epoch=0, batch=0, dataset='train', visualizations_dir = None)

visualization_utils.py:7

Method__init__

(self, norm_stats, camera_names)

robot_operation.py:80

Method__init__

(self)

robot_operation.py:183

Method__init__

(self)

robot_operation.py:212

(self, episode_ids, dataset_dir, camera_names, norm_stats, chunk_size, image_size = None)

utils.py:181

Method__init__

Create a projection head for CLIP. The projection head consists of an average pooling layer followed by a linear layer. out_

clip_pretraining.py:67

Method__init__

(self, state_dim: int, hidden_dim: int, position_embedding_

policy.py:27

Method__init__

(self, out_dim: int, conditioning_dim: int = 0, num_channels:int = 512, normailize: bool = True)

clip_pretraining_no_pos.py:61

Method__init__

(self, out_dim: int, conditioning_dim: int = 0, num_channels:int = 512, normailize: bool = True)

failed_DDP.py:72

Method__init__

(self, dim)

diffusion/network.py:33

Method__init__

(self, dim)

diffusion/network.py:48

Method__init__

(self, inp_channels, out_channels, kernel_size, n_groups=8)

diffusion/network.py:69

Method__init__

(self, in_channels, out_channels, cond_dim, kernel_size=3,

diffusion/network.py:83

Method__init__

input_dim: Dim of actions. global_cond_dim: Dim of global conditioning applied with FiLM in addition to diffusion step embe

diffusion/network.py:133

Method__init__

(self, norm_stats, camera_names)

diffusion/robot_operation.py:73

Method__init__

(self, norm_stats)

diffusion/utils.py:15

Method__init__

(self, norm_stats)

diffusion/utils.py:32

Method__init__

(self, episode_ids, dataset_dir, camera_names, norm_stats, chunk_size, image_size = None)

diffusion/utils.py:160

Method__init__

(self, out_dim: int, conditioning_dim: int = 0, num_channels:int = 512)

diffusion/clip_pretraining.py:75

Method__init__

(self, norm_stats)

diffusion/dataset.py:7

Method__init__

(self, episode_ids, dataset_dir, pred_horizon, camera_names,norm_stats,image_size=None)

diffusion/dataset.py:51

Method__init__

(self, norm_stats, camera_names)

diffusion/robot_operation_old.py:39

Method__init__

(self, print=False, plot=False, epoch=0, batch=0, dataset='train', visualizations_dir = None)

diffusion/visualization.py:6

Method__init__

(self, encoder_layer, num_layers, norm=None)

detr/detr/models/transformer.py:112

Method__init__

(self, decoder_layer, num_layers, norm=None, return_intermediate=False)

detr/detr/models/transformer.py:137

Method__init__

(self, d_model, nhead, dim_feedforward=2048, dropout=0.1, activation="relu", normalize_before

detr/detr/models/transformer.py:183

Method__init__

(self, d_model, nhead, dim_feedforward=2048, dropout=0.1, activation="relu", normalize_before

detr/detr/models/transformer.py:255

Method__init__

(self, n)

detr/detr/models/backbone.py:30

Method__init__

(self, backbone: nn.Module, train_backbone: bool, num_channels: int, return_interm_layers: bool)

detr/detr/models/backbone.py:62

Method__init__

(self, name: str, train_backbone: bool, return_interm_layers: bool,

detr/detr/models/backbone.py:78

Method__init__

(self, num_pos_feats=256)

detr/detr/models/position_encoding.py:59

Method__init__

(self, window_size=20, fmt=None)

detr/detr/util/misc.py:32

Method__init__

(self, delimiter="\t")

detr/detr/util/misc.py:160

Method__init__

(self, tensors, mask: Optional[Tensor])

detr/detr/util/misc.py:285

clip_pretraining.py:223

Method__len__

(self)

clip_pretraining_no_pos.py:205

diffusion/utils.py:79

Method__len__

(self)

diffusion/clip_pretraining.py:154

Method__repr__

(self)

detr/detr/util/misc.py:303

Method__str__

(self)

detr/detr/util/misc.py:80

Method__str__

(self)

detr/detr/util/misc.py:179

Method_load_from_state_dict

(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys

detr/detr/models/backbone.py:37

Functionaccuracy

Computes the precision@k for the specified values of k

detr/detr/util/misc.py:433

Methodadd_meter

(self, name, meter)

detr/detr/util/misc.py:191

Functionall_gather

Run all_gather on arbitrary picklable data (not necessarily tensors) Args: data: any picklable object Returns: list[data]

detr/detr/util/misc.py:89

Methodavg

(self)

detr/detr/util/misc.py:64

Functionbox_cxcywh_to_xyxy

(x)

detr/detr/util/box_ops.py:9

Functionbox_xyxy_to_cxcywh

(x)

detr/detr/util/box_ops.py:16

Functionbuild_ACT_model_and_optimizer

(args_override)

detr/detr/main.py:79

Functionbuild_CNNMLP_model_and_optimizer

(args_override)

detr/detr/main.py:102

Functionbuild_backbone

(args)

detr/detr/models/backbone.py:105

Functionbuild_position_encoding

(args)

detr/detr/models/position_encoding.py:83

Functionbuild_transformer

(args)

detr/detr/models/transformer.py:383

Functionclip_loss_old

(image_embeddings, gelsight_embeddings, target_matrix, logit_scale = 1.0, visualize = False)

clip_pretraining.py:328

Functionclip_loss_old

(image_embeddings, gelsight_embeddings, target_matrix, logit_scale = 1.0, visualize = False)

clip_pretraining_no_pos.py:319

Functionclip_loss_old

(image_embeddings, gelsight_embeddings, target_matrix, logit_scale = 1.0, visualize = False)

failed_DDP.py:335

Functionclip_pretraining

(train_loader: DataLoader, test_loader: DataLoader, device: torch.de

diffusion/clip_pretraining.py:253

Functioncollate_fn

(batch)

detr/detr/util/misc.py:269

Functioncompute_dict_mean

(epoch_dicts)

diffusion/utils.py:273

Methodconfigure_optimizers

detr/detr/util/misc.py:300

Functiondetach_dict

(d)

diffusion/utils.py:283

Methodforward

qpos: batch, qpos_dim image: batch, num_cam, channel, height, width env_state: None z: batch, z_dim, latent sample

detrvae.py:78

Methodforward

(self, feature_map, conditioning=None)

clip_pretraining.py:83

(self, feature_map, conditioning=None)

clip_pretraining_no_pos.py:69

← previous next →101–200 of 267, ranked by callers