Functions — github.com/MarkFzp/act-plus-plus

CodeHub MCPcopy Create free account

hub / github.com/MarkFzp/act-plus-plus / functions

Functions274 in github.com/MarkFzp/act-plus-plus

⨍ Functions 274◇ Types & classes 40

↓ 1 callersMethodget_qpos

↓ 1 callersMethodget_qpos

ee_sim_env.py:106

↓ 1 callersMethodget_qvel

↓ 1 callersMethodget_qvel

ee_sim_env.py:117

↓ 1 callersFunctionget_rank

detr/util/misc.py:392

↓ 1 callersFunctionget_sinusoid_encoding_table

(n_position, d_hid)

detr/models/detr_vae.py:24

↓ 1 callersFunctionkl_divergence

↓ 1 callersFunctionload_and_save_first_episode_video

(dataset_dir, video_path)

truncate_data.py:104

↓ 1 callersFunctionload_and_save_first_episode_video

(dataset_dir, video_path)

compress_data.py:128

↓ 1 callersFunctionload_hdf5

(dataset_dir, dataset_name)

visualize_episodes.py:16

↓ 1 callersFunctionload_hdf5

(dataset_dir, dataset_name)

truncate_data.py:110

↓ 1 callersFunctionload_hdf5

(dataset_dir, dataset_name)

compress_data.py:134

↓ 1 callersFunctionload_hdf5

(dataset_dir, dataset_name)

postprocess_episodes.py:21

↓ 1 callersFunctionmain

vinn_select_k.py:42

↓ 1 callersFunctionmain

visualize_episodes.py:33

↓ 1 callersFunctionmain

Generate demonstration data in simulation. First rollout the policy (defined in ee space) in ee_sim_env. Obtain the joint trajectory. Rep

record_sim_episodes.py:17

↓ 1 callersFunctionmain

postprocess_episodes.py:56

↓ 1 callersFunctionmain

vinn_eval.py:56

↓ 1 callersFunctionmain

vinn_cache_feature.py:29

↓ 1 callersFunctionmain

replay_episodes.py:15

↓ 1 callersFunctionmain

train_actuator_network.py:21

↓ 1 callersFunctionmain

↓ 1 callersFunctionmain

imitate_episodes.py:37

↓ 1 callersFunctionmain

train_latent_model.py:26

↓ 1 callersFunctionmake_optimizer

(policy_class, policy)

imitate_episodes.py:194

↓ 1 callersFunctionmake_policy

(policy_class, policy_config)

train_latent_model.py:128

↓ 1 callersFunctionmlp

(input_dim, hidden_dim, output_dim, hidden_depth)

detr/models/detr_vae.py:240

↓ 1 callersFunctionnested_tensor_from_tensor_list

(tensor_list: List[Tensor])

detr/util/misc.py:307

↓ 1 callersFunctionreparametrize

detr/models/detr_vae.py:18

↓ 1 callersMethodreset_parameters

detr/models/position_encoding.py:65

↓ 1 callersFunctionsave_videos

(video, dt, video_path=None)

compress_data.py:89

↓ 1 callersMethodset_torque_enabled

Sets whether torque is enabled for the motors. Args: motor_ids: The motor IDs to configure. enabled: Whether to engag

dynamixel_client.py:180

↓ 1 callersFunctionsetup_for_distributed

This function disables printing when not in master process

detr/util/misc.py:363

↓ 1 callersFunctionsigned_to_unsigned

Converts the given value to its unsigned representation.

dynamixel_client.py:43

↓ 1 callersFunctionsmooth_base_action

↓ 1 callersMethodsync_write

Writes values to a group of motors. Args: motor_ids: The motor IDs to write to. values: The values to write.

dynamixel_client.py:265

↓ 1 callersFunctiontest_policy

scripted_policy.py:152

↓ 1 callersFunctiontest_sim_teleop

Testing teleoperation in sim with ALOHA. Requires hardware and ALOHA repo to work.

↓ 1 callersFunctiontrain_bc

(train_dataloader, val_dataloader, config)

imitate_episodes.py:535

↓ 1 callersFunctiontrain_bc

(train_dataloader, val_dataloader, config, ckpt_name)

train_latent_model.py:346

↓ 1 callersMethodupdate

(self, **kwargs)

detr/util/misc.py:164

↓ 1 callersFunctionvisualize_joints

(qpos_list, command_list, plot_path=None, ylim=None, label_overwrite=None)

visualize_episodes.py:85

↓ 1 callersMethodvq_encode

(self, qpos, actions, is_pad)

↓ 1 callersMethodwrite_byte

Writes a value to the motors. Args: motor_ids: The motor IDs to write to. value: The value to write to the control ta

dynamixel_client.py:238

↓ 1 callersMethodwrite_desired_pos

Writes the given desired positions. Args: motor_ids: The motor IDs to write to. positions: The joint angles in radian

dynamixel_client.py:223

(self, qpos, image, actions=None, is_pad=None)

(self, qpos, image, actions=None, is_pad=None, vq_sample=None)

(self, qpos, image, actions=None, is_pad=None)

scripted_policy.py:36

Automatically disconnect on destruction.

dynamixel_client.py:344

Method__enter__

Enables use as a context manager.

dynamixel_client.py:334

Enables use as a context manager.

dynamixel_client.py:340

Method__getattr__

detr/util/misc.py:171

Initializes a new reader.

dynamixel_client.py:355

(self, client: DynamixelClient, motor_ids: Sequence[int], p

dynamixel_client.py:422

(self, client: DynamixelClient, motor_ids: Sequence[int], p

dynamixel_client.py:468

(self, client: DynamixelClient, motor_ids: Sequence[int], p

dynamixel_client.py:500

(self, client: DynamixelClient, motor_ids: Sequence[int], p

dynamixel_client.py:534

(self, random=None)

(self, random=None)

(self, dataset_path_list, camera_names, norm_stats, episode_ids, episode_len, chunk_size, policy_class)

(self, args_override)

(self, args_override)

(self, d_model: int, dropout: float = 0.1, max_len: int = 5000)

train_actuator_network.py:253

(self, dataset_path_list, norm_stats, episode_ids, episode_len, history_len, future_len, prediction_len)

train_actuator_network.py:307

(self, random=None)

ee_sim_env.py:55

(self, random=None)

ee_sim_env.py:154

(self, inject_noise=False)

scripted_policy.py:13

Initializes the model. Parameters: backbones: torch module of the backbone to be used. See backbone.py transformer: t

detr/models/detr_vae.py:37

(self, encoder_layer, num_layers, norm=None)

detr/models/transformer.py:81

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

detr/models/transformer.py:105

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

detr/models/transformer.py:146

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

detr/models/transformer.py:206

detr/models/backbone.py:30

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

detr/models/backbone.py:62

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

detr/models/backbone.py:88

(self, num_pos_feats=256)

detr/models/position_encoding.py:59

(self, input_dim, output_dim, seq_len, latent_dim=256, num_head=8, num_layer=3)

detr/models/latent_model.py:36

(self, window_size=20, fmt=None)

detr/util/misc.py:32

(self, delimiter="\t")

detr/util/misc.py:160

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

detr/util/misc.py:285

train_actuator_network.py:324

detr/util/misc.py:303

detr/util/misc.py:80

detr/util/misc.py:179

Method_get_data

Returns a copy of the data.

dynamixel_client.py:459

Method_get_data

Returns a copy of the data.

dynamixel_client.py:493

Method_get_data

Returns a copy of the data.

dynamixel_client.py:527

Method_get_data

Returns a copy of the data.

dynamixel_client.py:559

Method_initialize_data

Initializes the cached data.

dynamixel_client.py:438

Method_initialize_data

Initializes the cached data.

dynamixel_client.py:482

Method_initialize_data

Initializes the cached data.

dynamixel_client.py:516

Method_initialize_data

Initializes the cached data.

dynamixel_client.py:548

Method_load_from_state_dict

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

detr/models/backbone.py:37

Method_update_data

Updates the data index for the given motor ID.

dynamixel_client.py:444

Method_update_data

Updates the data index for the given motor ID.

dynamixel_client.py:486

Method_update_data

Updates the data index for the given motor ID.

dynamixel_client.py:520

Method_update_data

Updates the data index for the given motor ID.

dynamixel_client.py:552

Functionaccuracy

Computes the precision@k for the specified values of k

detr/util/misc.py:433

Methodadd_meter

(self, name, meter)

detr/util/misc.py:191

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