MCPcopy Create free account

hub / github.com/TencentARC/Moto / functions

Functions210 in github.com/TencentARC/Moto

⨍ Functions 210◇ Types & classes 46
↓ 7 callersFunctionparse_range_tuple
(t)
moto_gpt/evaluation/robot_manipulation_benchmarks/simpler/evaluate_simpler.py:13
↓ 6 callersMethod__init__
(self, n_state, config)
moto_gpt/src/models/trajectory_gpt2.py:419
↓ 6 callersMethodnext
(self)
common/data/datasets.py:68
↓ 6 callersMethodprint
(self, *args, **kwargs)
moto_gpt/src/trainers/moto_gpt_trainer.py:119
↓ 5 callersMethod__init__
(self, loader, device, lang_tokenizer=None)
common/data/datasets.py:35
↓ 5 callersMethodpost_process
(self, x)
common/processors/rgb_preprocessors.py:86
↓ 4 callersMethoddecode_image
(self, cond_pixel_values, given_motion_token_ids)
latent_motion_tokenizer/src/models/latent_motion_tokenizer.py:77
↓ 4 callersMethodget_last_lr
(self)
latent_motion_tokenizer/src/trainers/optimizer.py:72
↓ 4 callersFunctionload_model
(pretrained_path)
common/models/model_utils.py:15
↓ 4 callersMethodlog
(self, log_loss, eval_log_loss, cum_load_time, clock, epoch, batch_idx, step)
moto_gpt/src/trainers/moto_gpt_trainer.py:406
↓ 4 callersMethodnext_without_none
(self)
common/data/datasets.py:78
↓ 3 callersFunctionmasked_loss
(pred, target, mask, skip_frame=0, loss_func=F.mse_loss)
moto_gpt/src/trainers/trainer_utils.py:16
↓ 3 callersMethodprint
(self, msg)
latent_motion_tokenizer/src/trainers/latent_motion_tokenizer_trainer.py:107
↓ 3 callersMethodprune_heads
(self, heads)
moto_gpt/src/models/trajectory_gpt2.py:316
↓ 3 callersMethodreset
(self, task_description: str)
moto_gpt/evaluation/robot_manipulation_benchmarks/simpler/simpler_env_policy_wrapper.py:108
↓ 3 callersMethodsplit_heads
(self, x, k=False)
moto_gpt/src/models/trajectory_gpt2.py:365
↓ 2 callersMethod__init__
(self, config: ViTConfig)
latent_motion_tokenizer/src/models/modeling_lmd_vit.py:73
↓ 2 callersMethod__init__
(self, config: ViTConfig, add_pooling_layer: bool = True)
latent_motion_tokenizer/src/models/m_former.py:76
↓ 2 callersMethod_rescale_action_with_bound
Formula taken from https://stats.stackexchange.com/questions/281162/scale-a-number-between-a-range.
moto_gpt/evaluation/robot_manipulation_benchmarks/simpler/simpler_env_policy_wrapper.py:64
↓ 2 callersMethodcalculate_loss
(self, batch, train)
latent_motion_tokenizer/src/trainers/latent_motion_tokenizer_trainer.py:233
↓ 2 callersMethodcalculate_loss
(self, batch, train)
moto_gpt/src/trainers/moto_gpt_trainer.py:348
↓ 2 callersFunctioncount_success
(results)
moto_gpt/evaluation/robot_manipulation_benchmarks/calvin/calvin_utils.py:29
↓ 2 callersMethoddevice
(self)
moto_gpt/src/models/moto_gpt.py:138
↓ 2 callersMethodeval_latent_motion_gen
(self, visualization_dir)
moto_gpt/src/trainers/moto_gpt_trainer.py:223
↓ 2 callersMethodeval_latent_motion_reconstruction
(self, visualization_dir)
latent_motion_tokenizer/src/trainers/latent_motion_tokenizer_trainer.py:203
↓ 2 callersFunctionget_model_vision_basic_config
(model_vision_type)
common/processors/preprocessor_utils.py:5
↓ 2 callersMethodget_npz_path
(self, npz_basename)
common/data/datasets.py:498
↓ 2 callersFunctionget_rgb_preprocessor
(model_vision_type, vision_aug_config={})
common/processors/preprocessor_utils.py:30
↓ 2 callersFunctionget_split_and_ratio
(split, splits)
common/data/datasets.py:18
↓ 2 callersFunctionload_dataset
(data_config, extra_data_config)
common/data/data_utils.py:25
↓ 2 callersFunctionload_moto_gpt_policy
(args)
common/models/model_utils.py:29
↓ 2 callersMethodpreload
(self)
common/data/datasets.py:43
↓ 2 callersMethodreset
( self, reset_info: Dict[str, Any] = None, batch_idx: int = 0, seq_idx: int =
moto_gpt/evaluation/robot_manipulation_benchmarks/calvin/calvin_env_wrapper_raw.py:96
↓ 2 callersFunctionrun_maniskill2_eval_single_episode
( model, robot_name, env_name, scene_name, robot_init_x, robot_init_y, robot_init_
moto_gpt/evaluation/robot_manipulation_benchmarks/simpler/maniskill2_evaluator.py:15
↓ 2 callersMethodsave_checkpoint
(self, save_dir)
latent_motion_tokenizer/src/trainers/latent_motion_tokenizer_trainer.py:110
↓ 2 callersMethodsave_checkpoint
(self, save_dir)
moto_gpt/src/trainers/moto_gpt_trainer.py:122
↓ 2 callersFunctionsave_to_lmdb
(output_dir, input_dir)
data_preprocessing/calvin_to_lmdb.py:20
↓ 2 callersMethodstep
(self)
latent_motion_tokenizer/src/trainers/optimizer.py:62
↓ 2 callersMethodstep
Step function.
moto_gpt/src/models/moto_gpt_policy_wraper.py:80
↓ 2 callersMethodstep
( self, action_tensor: torch.Tensor )
moto_gpt/evaluation/robot_manipulation_benchmarks/calvin/calvin_env_wrapper_raw.py:74
↓ 2 callersMethodstep
Input: image: np.ndarray of shape (H, W, 3), uint8 task_description: Optional[str], task description; if different fr
moto_gpt/evaluation/robot_manipulation_benchmarks/simpler/simpler_env_policy_wrapper.py:148
↓ 1 callersMethod__init__
(self, last_hidden_state)
latent_motion_tokenizer/src/models/theia_model.py:45
↓ 1 callersMethod__init__
(self, last_hidden_state)
latent_motion_tokenizer/src/models/timm_dinov2_model.py:9
↓ 1 callersMethod__init__
(self, pad)
common/processors/rgb_preprocessors.py:10
↓ 1 callersMethod__init__
(self, last_hidden_state)
moto_gpt/src/models/theia_model.py:45
↓ 1 callersMethod__init__
(self, last_hidden_state)
moto_gpt/src/models/timm_dinov2_model.py:11
↓ 1 callersMethod_attn
(self, q, k, v, attention_mask=None, head_mask=None, output_attentions=False)
moto_gpt/src/models/trajectory_gpt2.py:333
↓ 1 callersMethod_initialize_task_description
(self, task_description: Optional[str] = None)
moto_gpt/evaluation/robot_manipulation_benchmarks/simpler/simpler_env_policy_wrapper.py:102
↓ 1 callersMethod_resize_image
(self, image: np.ndarray | tf.Tensor)
moto_gpt/evaluation/robot_manipulation_benchmarks/simpler/simpler_env_policy_wrapper.py:97
↓ 1 callersMethod_small_action_filter_google_robot
(raw_action: dict[str, np.ndarray], arm_movement: bool = False, gripper: bool = True)
moto_gpt/evaluation/robot_manipulation_benchmarks/simpler/simpler_env_policy_wrapper.py:116
↓ 1 callersMethodbeam_search_decode
(self, act_x, stacked_inputs, stacked_attention_mask, latent_motion_query_token_start_i, n_cond_tokens,
moto_gpt/src/models/moto_gpt.py:425
↓ 1 callersFunctioncompute_metrics
(eval_dir)
moto_gpt/evaluation/robot_manipulation_benchmarks/simpler/compute_metrics.py:8
↓ 1 callersMethoddecode_latent_motion
(self, act_x, stacked_inputs, stacked_attention_mask, latent_motion_query_token_start_i, n_cond_tokens,
moto_gpt/src/models/moto_gpt.py:341
↓ 1 callersFunctionevaluate_policy
(model, env, eval_sr_path, eval_result_path, ep_len, num_sequences, num_procs, procs_id, eval_dir=None, debug=
moto_gpt/evaluation/robot_manipulation_benchmarks/calvin/evaluate_calvin.py:72
↓ 1 callersFunctionevaluate_sequence
(env, model, task_checker, initial_state, eval_sequence, val_annotations, debug, eval_dir, sequence_i, ep_len)
moto_gpt/evaluation/robot_manipulation_benchmarks/calvin/evaluate_calvin.py:108
↓ 1 callersMethodextract_action
(self, idx)
common/data/datasets.py:195
↓ 1 callersMethodextract_action
(self, idx)
common/data/datasets.py:314
↓ 1 callersMethodextract_actions
(self, idx, cur_episode, delta_t, actions, mask)
common/data/datasets.py:186
↓ 1 callersMethodextract_frames
(self, idx, cur_episode, delta_t, rgb_initial, rgb_future, latent_mask)
common/data/datasets.py:155
↓ 1 callersMethodextract_frames
(self, video_basename, start_local_step, num_frames, delta_t, rgb_initial, rgb_future,
common/data/datasets.py:370
↓ 1 callersMethodextract_frames
(self, npz_basename, delta_t, rgb_initial, rgb_future, latent_mask)
common/data/datasets.py:501
↓ 1 callersMethodextract_lang_goal
(self, idx, cur_episode)
common/data/datasets.py:146
↓ 1 callersFunctionget_args
()
moto_gpt/evaluation/robot_manipulation_benchmarks/simpler/evaluate_simpler.py:16
↓ 1 callersMethodget_codebook_entry
(self, indices)
latent_motion_tokenizer/src/models/vector_quantizer.py:124
↓ 1 callersFunctionget_dataset_path
(parent_dir, dataset_name)
data_preprocessing/oxe_to_video.py:15
↓ 1 callersFunctionget_dataset_path
(parent_dir, dataset_name)
data_preprocessing/oxe_to_lmdb.py:17
↓ 1 callersFunctionget_image_processor
(vision_processor_config)
moto_gpt/evaluation/motion_prior_analysis/latent_motion_generation.py:21
↓ 1 callersMethodget_info
(self)
moto_gpt/evaluation/robot_manipulation_benchmarks/calvin/calvin_env_wrapper_raw.py:117
↓ 1 callersMethodget_obs
(self)
moto_gpt/evaluation/robot_manipulation_benchmarks/calvin/calvin_env_wrapper_raw.py:120
↓ 1 callersFunctionget_optimizer
( params, lr=1e-4, wd=1e-2, betas=(0.9, 0.99), eps=1e-8, filter_by_requires_grad
latent_motion_tokenizer/src/trainers/optimizer.py:12
↓ 1 callersMethodget_state_dict_to_save
(self)
latent_motion_tokenizer/src/models/latent_motion_tokenizer.py:69
↓ 1 callersMethodget_state_dict_to_save
(self)
moto_gpt/src/models/moto_gpt.py:141
↓ 1 callersMethodget_video_path
(self, cur_episode)
common/data/datasets.py:151
↓ 1 callersMethodget_video_path
(self, video_basename)
common/data/datasets.py:367
↓ 1 callersMethodgreedy_decode
(self, act_x, stacked_inputs, stacked_attention_mask, latent_motion_query_token_start_i, n_cond_tokens, parall
moto_gpt/src/models/moto_gpt.py:373
↓ 1 callersFunctionhandle_feature_output
Handle feature output from transformer. Args: x (torch.Tensor): input feature to be handled. shape is [B, 1+H*W+N, C] if inclu
latent_motion_tokenizer/src/models/theia_model.py:6
↓ 1 callersFunctionhandle_feature_output
Handle feature output from transformer. Args: x (torch.Tensor): input feature to be handled. shape is [B, 1+H*W+N, C] if inclu
moto_gpt/src/models/theia_model.py:6
↓ 1 callersFunctioninference
( moto_gpt, latent_motion_tokenizer, lang_tokenizer, image_processor,
moto_gpt/evaluation/motion_prior_analysis/latent_motion_generation.py:95
↓ 1 callersMethodlog
(self, log_loss, eval_log_loss, cum_load_time, clock, epoch, batch_idx, step)
latent_motion_tokenizer/src/trainers/latent_motion_tokenizer_trainer.py:247
↓ 1 callersFunctionmain
(cfg)
latent_motion_tokenizer/train/train_latent_motion_tokenizer.py:17
↓ 1 callersFunctionmain
(args)
moto_gpt/evaluation/robot_manipulation_benchmarks/calvin/evaluate_calvin.py:161
↓ 1 callersFunctionmain
(args)
moto_gpt/evaluation/motion_prior_analysis/latent_motion_generation.py:246
↓ 1 callersFunctionmain
(cfg)
moto_gpt/train/train_moto_gpt.py:17
↓ 1 callersFunctionmake_env
(dataset_path, observation_space, device)
moto_gpt/evaluation/robot_manipulation_benchmarks/calvin/evaluate_calvin.py:65
↓ 1 callersFunctionmaniskill2_evaluator
(model, args)
moto_gpt/evaluation/robot_manipulation_benchmarks/simpler/maniskill2_evaluator.py:175
↓ 1 callersMethodmerge_heads
(self, x)
moto_gpt/src/models/trajectory_gpt2.py:360
↓ 1 callersMethodobtain_item
(self, idx, start_local_step=None, delta_t=None)
common/data/datasets.py:401
↓ 1 callersMethodobtain_item
(self, idx, delta_t=None)
common/data/datasets.py:530
↓ 1 callersMethodopen_lmdb
(self)
common/data/datasets.py:142
↓ 1 callersFunctionprint_and_save
(results, sequences, eval_result_path, epoch=None)
moto_gpt/evaluation/robot_manipulation_benchmarks/calvin/calvin_utils.py:39
↓ 1 callersMethodremap_to_used
(self, inds)
latent_motion_tokenizer/src/models/vector_quantizer.py:55
↓ 1 callersMethodrgb_process
(self, rgb)
moto_gpt/src/models/moto_gpt_policy_wraper.py:63
↓ 1 callersFunctionrollout
(env, model, task_oracle, subtask, val_annotations, debug, eval_dir, subtask_i, sequence_i, ep_len)
moto_gpt/evaluation/robot_manipulation_benchmarks/calvin/evaluate_calvin.py:127
↓ 1 callersFunctionsave_to_lmdb
(output_dir, input_dir, dataset_name)
data_preprocessing/oxe_to_lmdb.py:30
↓ 1 callersFunctionseparate_weight_decayable_params
(params)
latent_motion_tokenizer/src/trainers/optimizer.py:4
↓ 1 callersMethodset_egl_device
(device)
moto_gpt/evaluation/robot_manipulation_benchmarks/calvin/calvin_env_wrapper_raw.py:57
↓ 1 callersMethodsetup_remap
(self, remap, unknown_index)
latent_motion_tokenizer/src/models/vector_quantizer.py:43
↓ 1 callersMethodtokenize
(self, cond_pixel_values, target_pixel_values, before_vq=False)
latent_motion_tokenizer/src/models/latent_motion_tokenizer.py:99
↓ 1 callersFunctiontop_k_top_p_filtering
Filter a distribution of logits using top-k and/or nucleus (top-p) filtering Args: logits: logits distribution shape (batch size,
moto_gpt/src/models/moto_gpt.py:530
↓ 1 callersMethodtrain
(self)
latent_motion_tokenizer/src/trainers/latent_motion_tokenizer_trainer.py:119
next →1–100 of 210, ranked by callers