Functions — github.com/bytedance/GR-MG

hub / github.com/bytedance/GR-MG / functions

Functions177 in github.com/bytedance/GR-MG

⨍ Functions 177◇ Types & classes 30 ↳ Endpoints 1

↓ 23 callersFunctionprint

(*argc, all=False, **kwargs)

policy/utils/dist_train.py:31

↓ 22 callersMethodload

(self,path)

policy/data/calvin_dataset.py:156

↓ 9 callersFunctionprint

(*argc, all=False, **kwargs)

goal_gen/utils/dist_train.py:31

↓ 8 callersMethod_main_rank_print

(*args, **kwargs)

policy/training/trainer.py:67

↓ 6 callersMethod__init__

(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)

policy/model/vision_transformer.py:139

↓ 6 callersFunctioneuler2rotm

Euler angle (ZYX) to rotation matrix.

policy/data/utils.py:44

↓ 4 callersMethod__init__

(self, intermediate_size, config)

policy/model/gpt2.py:345

↓ 4 callersMethod_main_rank_print

(*args, **kwargs)

goal_gen/training/trainer.py:31

↓ 4 callersFunctiontrunc_normal_

(tensor, mean=0., std=1., a=-2., b=2.)

policy/model/vision_transformer.py:112

↓ 3 callersMethod_split_heads

Splits hidden_size dim into attn_head_size and num_heads

policy/model/gpt2.py:274

↓ 3 callersFunctioneuler2rotm

Euler angle (ZYX) to rotation matrix.

evaluate/utils/utils.py:106

↓ 3 callersMethodforward

(self, input_dict)

goal_gen/model/model.py:67

↓ 3 callersFunctionis_dist

()

policy/utils/dist_train.py:28

↓ 3 callersFunctionis_dist

()

goal_gen/utils/dist_train.py:28

↓ 3 callersMethodprepare_tokens

(self, x)

policy/model/vision_transformer.py:285

↓ 3 callersFunctionrotm2euler

(R)

policy/data/utils.py:67

↓ 3 callersMethodstep

Step function.

evaluate/wrapper/model_wrapper.py:145

↓ 2 callersMethod__init__

(self, pad)

policy/data/ego4d_dataset.py:37

↓ 2 callersMethod__init__

(self, pad)

policy/data/calvin_dataset.py:32

↓ 2 callersMethod_log_output

(self, output, phase, dataset=None, **kwargs)

policy/training/trainer.py:216

↓ 2 callersMethod_log_output

(self, output, phase, dataset=None, **kwargs)

goal_gen/training/trainer.py:89

↓ 2 callersFunctioncompute_kl_divergence

( mu, logvar)

policy/training/trainer.py:38

↓ 2 callersFunctioncount_success

(results)

evaluate/utils/utils.py:30

↓ 2 callersFunctiondeep_update

(d1, d2)

goal_gen/train.py:127

↓ 2 callersFunctionget_1d_sincos_pos_embed_from_grid

embed_dim: output dimension for each position pos: a list of positions to be encoded: size (M,) out: (M, D)

policy/model/vision_transformer.py:26

↓ 2 callersFunctionget_date_str

()

policy/main.py:42

↓ 2 callersFunctionget_date_str

()

goal_gen/train.py:34

↓ 2 callersFunctionget_rank

()

policy/utils/dist_train.py:24

↓ 2 callersMethodinference

Inference function.

goal_gen/evaluate.py:116

↓ 2 callersFunctionnorm_cdf

(x)

policy/model/vision_transformer.py:80

↓ 2 callersFunctionprint_and_save

(results, sequences, eval_result_path, epoch=None)

evaluate/utils/utils.py:39

↓ 2 callersMethodreset

Reset function.

evaluate/wrapper/model_wrapper.py:118

↓ 2 callersFunctionrotm2euler

(R)

evaluate/utils/utils.py:129

↓ 2 callersMethodtokenize_texts

(self, texts)

goal_gen/model/model.py:63

↓ 1 callersMethod_attn

(self, query, key, value, attention_mask=None, head_mask=None)

policy/model/gpt2.py:182

↓ 1 callersMethod_encode_prompt

r""" Encodes the prompt into text encoder hidden states. Args: prompt (`str` or `List[str]`, *optional*):

goal_gen/utils/pipeline.py:63

↓ 1 callersMethod_initialize

Generate the sequence index pair.

policy/data/calvin_dataset.py:172

↓ 1 callersMethod_initialize

(self)

policy/training/trainer.py:75

↓ 1 callersMethod_initialize

(self)

goal_gen/training/trainer.py:39

↓ 1 callersMethod_merge_heads

Merges attn_head_size dim and num_attn_heads dim into hidden_size

policy/model/gpt2.py:282

↓ 1 callersFunction_no_grad_trunc_normal_

(tensor, mean, std, a, b)

policy/model/vision_transformer.py:77

↓ 1 callersMethod_upcast_and_reordered_attn

(self, query, key, value, attention_mask=None, head_mask=None)

policy/model/gpt2.py:222

↓ 1 callersFunctionalpha2rotm

Alpha euler angle to rotation matrix.

policy/data/utils.py:17

↓ 1 callersFunctionalpha2rotm

Alpha euler angle to rotation matrix.

evaluate/utils/utils.py:79

↓ 1 callersFunctionbeta2rotm

Beta euler angle to rotation matrix.

policy/data/utils.py:26

↓ 1 callersFunctionbeta2rotm

Beta euler angle to rotation matrix.

evaluate/utils/utils.py:88

↓ 1 callersFunctioncheck_no_lan_frame

(path)

policy/data/generate_play.py:18

↓ 1 callersMethodcompute_rel_state

(states)

evaluate/wrapper/model_wrapper.py:126

↓ 1 callersFunctiondrop_path

(x, drop_prob: float = 0., training: bool = False)

policy/model/vision_transformer.py:116

↓ 1 callersMethodencode_texts

(self, texts)

policy/model/model.py:215

↓ 1 callersMethodevaluate

(self, input_dict,original_gripper=False,return_progress=False)

policy/model/model.py:576

↓ 1 callersMethodevaluate

(self, eval_result_dir, eval_data_dir,is_training)

goal_gen/evaluate.py:70

↓ 1 callersFunctionevaluate_policy

Run this function to evaluate a model on the CALVIN challenge.

evaluate/eval.py:68

↓ 1 callersFunctionevaluate_sequence

Evaluates a sequence of language instructions.

evaluate/eval.py:95

↓ 1 callersFunctionexperiment

(variant)

policy/main.py:96

↓ 1 callersFunctionexperiment

(variant)

goal_gen/train.py:82

↓ 1 callersMethodforward

(self, input_dict, is_training=True)

policy/model/model.py:222

↓ 1 callersFunctiongamma2rotm

Gamma euler angle to rotation matrix.

policy/data/utils.py:35

↓ 1 callersFunctiongamma2rotm

Gamma euler angle to rotation matrix.

evaluate/utils/utils.py:97

↓ 1 callersFunctionget_2d_sincos_pos_embed

grid_size: int of the grid height and width return: pos_embed: [grid_size*grid_size, embed_dim] or [1+grid_size*grid_size, embed_dim] (w/

policy/model/vision_transformer.py:58

↓ 1 callersFunctionget_2d_sincos_pos_embed_from_grid

(embed_dim, grid)

policy/model/vision_transformer.py:47

↓ 1 callersMethodget_annotations

(self)

policy/data/ego4d_dataset.py:136

↓ 1 callersMethodget_info

(self)

evaluate/wrapper/calvin_env_wrapper_raw.py:111

↓ 1 callersFunctionget_local_rank

()

policy/utils/dist_train.py:25

↓ 1 callersFunctionget_local_rank

()

goal_gen/utils/dist_train.py:25

↓ 1 callersMethodget_obs

(self)

evaluate/wrapper/calvin_env_wrapper_raw.py:114

↓ 1 callersFunctionget_rank

()

goal_gen/utils/dist_train.py:24

↓ 1 callersFunctionget_world_size

()

policy/utils/dist_train.py:23

↓ 1 callersFunctionget_world_size

()

goal_gen/utils/dist_train.py:23

↓ 1 callersFunctioninit_setup_callback

(config)

policy/main.py:58

↓ 1 callersFunctioninit_setup_callback

(config)

goal_gen/train.py:38

↓ 1 callersFunctioninit_trainer_config

(configs)

policy/main.py:65

↓ 1 callersFunctioninit_trainer_config

(configs)

goal_gen/train.py:45

↓ 1 callersMethodinterpolate_pos_encoding

(self, x, w, h)

policy/model/vision_transformer.py:263

↓ 1 callersFunctionisRotm

(R)

policy/data/utils.py:58

↓ 1 callersFunctionisRotm

(R)

evaluate/utils/utils.py:120

↓ 1 callersFunctionload_config

(config_file)

goal_gen/train.py:137

↓ 1 callersFunctionmain

()

evaluate/eval.py:157

↓ 1 callersFunctionmain

(data_dir, target_dir, num_trajs_per_task=10000)

goal_gen/utils/format_calvin_data.py:27

↓ 1 callersFunctionmake_env

(dataset_path, observation_space, device_id)

evaluate/eval.py:59

↓ 1 callersFunctionparse_args

()

policy/main.py:154

↓ 1 callersFunctionparse_args

()

goal_gen/train.py:160

↓ 1 callersMethodprune_heads

(self, heads)

policy/model/gpt2.py:167

↓ 1 callersFunctionreparameterize

(mu, logvar)

policy/model/model.py:27

↓ 1 callersFunctionrequires_grad

Set requires_grad flag for all parameters in a model.

goal_gen/utils/ema.py:29

↓ 1 callersFunctionrollout

Run the actual rollout on one subtask.

evaluate/eval.py:108

↓ 1 callersFunctionsave_play_traj

(path,file_number,save_name="play.json")

policy/data/generate_play.py:42

↓ 1 callersFunctionset_seed

(seed=0)

policy/main.py:34

↓ 1 callersFunctionset_seed

(seed=0)

goal_gen/train.py:74

↓ 1 callersFunctionupdate_configs

(configs, args)

goal_gen/train.py:145

↓ 1 callersFunctionupdate_ema

Step the EMA model towards the current model.

goal_gen/utils/ema.py:18

Method__call__

( self, prompt: Union[str, List[str]] = None, image =None, num_inference_steps

goal_gen/utils/pipeline.py:211

Method__getitem__

(self, index)

policy/data/ego4d_dataset.py:148

Method__getitem__

(self, index)

policy/data/calvin_dataset.py:223

Method__getitem__

(self, index)

goal_gen/data/calvindataset.py:90

Method__init__

(self, now, logdir, ckptdir)

policy/main.py:46

Method__init__

(self, config, is_cross_attention=False, layer_idx=None)

policy/model/gpt2.py:124

Method__init__

(self, config, layer_idx=None)

policy/model/gpt2.py:362

Method__init__

(self, *inputs, **kwargs)

policy/model/gpt2.py:453

Method__init__

(self, config)

policy/model/gpt2.py:670

next →1–100 of 177, ranked by callers