MCPcopy Create free account

hub / github.com/YvanYin/DrivingWorld / functions

Functions280 in github.com/YvanYin/DrivingWorld

↓ 1 callersFunctionmain
(args)
tools/test_change_road_demo.py:118
↓ 1 callersMethodmodel_forward
(self, feature_total, pose_indices_total, yaw_indices_total, targets, posedrop_input_flag, posedrop_gt_flag)
models/model.py:61
↓ 1 callersMethodnormalize_imgs
(self, imgs)
datasets/dataset_demo.py:22
↓ 1 callersMethodnormalize_imgs
(self, imgs)
datasets/dataset_nuplan.py:44
↓ 1 callersFunctionpk_sampling
(logits, top_k=30, temperature_k=1.0, top_p=0.9, temperature_p=1.0, filter_value=-float('Inf'))
utils/generat_utils.py:47
↓ 1 callersFunctionradians_to_degrees
(radians)
datasets/datasets_utils.py:31
↓ 1 callersFunctionreshape_for_broadcast
(freqs_cis: torch.Tensor, x: torch.Tensor)
modules/tokenizers/vq_model.py:569
↓ 1 callersFunctionreshape_for_broadcast
(freqs_cis, x)
utils/rope_2d.py:16
↓ 1 callersMethodstep_eval
(self, token, feature, pose, yaw, token_gt, feature_gt, pose_gt, yaw_gt, vqvae_codebook)
models/model.py:122
↓ 1 callersMethodstep_train
(self, token, feature, pose, yaw, token_gt, feature_gt, pose_gt, yaw_gt, vqvae_codebook=None)
models/model.py:84
↓ 1 callersFunctiontest_sliding_window_img
(val_data, model, args, tokenizer)
tools/test_long_term_demo.py:44
↓ 1 callersFunctiontest_sliding_window_img
(val_data, model, args, tokenizer)
tools/test_change_road_demo.py:44
↓ 1 callersMethodto_dict
Convert the ConfigDict to a normal dictionary recursively, and convert the ``LazyObject`` or ``LazyAttr`` to string.
utils/config_utils.py:845
↓ 1 callersFunctiontop_k_sampling
Sampling from top-k. In LLM, lowering the temperature can reduce the semantic inconsistency problem in long-context generation. Args:
utils/generat_utils.py:4
↓ 1 callersFunctiontop_p_sampling
Keep the top tokens with cumulative probability >= top_p (nucleus filtering), see https://arxiv.org/abs/1904.09751. In LLM, lowering the temp
utils/generat_utils.py:21
FunctionVQ_16
(**kwargs)
modules/tokenizers/vq_model.py:670
FunctionVQ_8
(**kwargs)
modules/tokenizers/vq_model.py:667
FunctionVideoVQ_16
(**kwargs)
modules/tokenizers/vq_model.py:673
Method__call__
(self, *args, **kwargs: Any)
utils/config_utils.py:121
Method__call__
(self, *args, **kwargs)
utils/config_utils.py:537
Method__copy__
(self)
utils/config_utils.py:681
Method__copy__
(self)
utils/config_utils.py:2034
Method__deepcopy__
(self, memo)
utils/config_utils.py:127
Method__deepcopy__
(self, memo)
utils/config_utils.py:540
Method__deepcopy__
(self, memo)
utils/config_utils.py:674
Method__deepcopy__
(self, memo)
utils/config_utils.py:2024
Method__eq__
(self, other)
utils/config_utils.py:818
Method__getattr__
(self, name: str)
utils/config_utils.py:124
Method__getattr__
(self, name)
utils/config_utils.py:543
Method__getattr__
(self, name)
utils/config_utils.py:641
Method__getitem__
(self, key)
utils/config_utils.py:671
Method__getitem__
(self, index)
datasets/dataset_demo.py:62
Method__getitem__
(self, index)
datasets/dataset_nuplan.py:104
Method__getstate__
(self)
utils/config_utils.py:154
Method__getstate__
(self)
utils/config_utils.py:563
Method__getstate__
( self )
utils/config_utils.py:2016
Method__init__
(self, config: ModelArgs)
modules/tokenizers/vq_model.py:25
Method__init__
(self, in_channels=3, ch=128, ch_mult=(1,1,2,2,4), num_res_blocks=2, norm_type='group', drop
modules/tokenizers/vq_model.py:160
Method__init__
(self, z_channels=256, ch=128, ch_mult=(1,1,2,2,4), num_res_blocks=2, norm_type="group", drop
modules/tokenizers/vq_model.py:208
Method__init__
(self, n_e, e_dim, beta, entropy_loss_ratio, l2_norm, show_usage)
modules/tokenizers/vq_model.py:260
Method__init__
(self, in_channels, out_channels=None, conv_shortcut=False, dropout=0.0, norm_type='group')
modules/tokenizers/vq_model.py:358
Method__init__
(self, in_channels, norm_type='group')
modules/tokenizers/vq_model.py:396
Method__init__
(self, n_heads, dim, multiple_of=256, ffn_dim_multiplier=None, norm_eps=1e-5)
modules/tokenizers/vq_model.py:433
Method__init__
(self, n_heads, dim)
modules/tokenizers/vq_model.py:461
Method__init__
( self, dim: int, hidden_dim: int, multiple_of: int, ffn_dim_multiplie
modules/tokenizers/vq_model.py:529
Method__init__
(self, dim: int, eps: float = 1e-6)
modules/tokenizers/vq_model.py:557
Method__init__
(self, in_channels, with_conv)
modules/tokenizers/vq_model.py:617
Method__init__
(self, in_channels, with_conv)
modules/tokenizers/vq_model.py:631
Method__init__
(self, args, pose_codebook_size=4096)
modules/tokenizers/model_tokenizer.py:60
Method__init__
(self, key)
utils/config_utils.py:38
Method__init__
(self, name: str, source: Union['LazyObject', 'LazyAttr'],
utils/config_utils.py:68
Method__init__
(self, global_dict: dict, base_dict: Optional[dict] = None,
utils/config_utils.py:278
Method__init__
(self, module: Union[str, list, tuple], imported: Optional[str] = None,
utils/config_utils.py:476
Method__init__
(__self, *args, **kwargs)
utils/config_utils.py:613
Method__init__
( self, src_dataset_ls, batch_size, rank, seed, num_repli
utils/merge_dataset.py:38
Method__init__
(self, data_root, condition_frames=15, downsample_size=16, h=256, w=512)
datasets/dataset_demo.py:10
Method__init__
(self, data_root, json_root, condition_frames=3, downsample_fps=5, downsample_size=16, h=256, w=512)
datasets/dataset_nuplan.py:12
Method__init__
(self, vocab_size, block_size, **kwargs)
models/ar.py:13
Method__init__
(self, config)
models/ar.py:20
Method__init__
(self, config)
models/ar.py:68
Method__init__
(self, config)
models/ar.py:87
Method__init__
(self, config)
models/ar.py:134
Method__init__
(self, config)
models/ar.py:153
Method__init__
(self, config)
models/ar.py:185
Method__init__
(self, config)
models/ar.py:204
Method__init__
( self, args, local_rank=-1, condition_frames=3,)
models/model.py:12
Method__iter__
(self)
utils/config_utils.py:689
Method__iter__
(self)
utils/config_utils.py:2013
Method__iter__
_summary_ Yields: list(int): a batch of indics, corresponding to ConcatDataset of src_dataset_ls
utils/merge_dataset.py:112
Method__len__
(self)
utils/config_utils.py:1994
Method__len__
(self)
utils/merge_dataset.py:135
Method__len__
(self)
datasets/dataset_demo.py:19
Method__len__
(self)
datasets/dataset_nuplan.py:31
Method__missing__
(self, name)
utils/config_utils.py:638
Method__reduce_ex__
(self, proto)
utils/config_utils.py:805
Method__repr__
(self)
utils/config_utils.py:1991
Method__setattr__
(self, name, value)
utils/config_utils.py:663
Method__setitem__
(self, name, value)
utils/config_utils.py:667
Method__setstate__
(self, state)
utils/config_utils.py:157
Method__setstate__
(self, state)
utils/config_utils.py:566
Method__setstate__
(self, state: Tuple[dict, Optional[str], Optional[str], dict, bool, se
utils/config_utils.py:2044
Method__str__
(self)
utils/config_utils.py:145
Method__str__
(self)
utils/config_utils.py:552
Method_contain_invalid_identifier
(dict_str)
utils/config_utils.py:1935
Method_format_basic_types
(k, v, use_mapping=False)
utils/config_utils.py:1889
Method_format_dict
(input_dict, outest_level=False)
utils/config_utils.py:1942
Method_format_list_tuple
(k, v, use_mapping=False)
utils/config_utils.py:1904
Method_get_base_module_from_with
Get base module name from if statement in python file. Args: with_nodes (list): List of if statement. Return
utils/config_utils.py:1053
Method_indent
(s_, num_spaces)
utils/config_utils.py:1879
Function_init_dist_envi
(args)
utils/comm.py:18
Method_init_weights
(self, module)
models/ar.py:334
Method_merge_a_into_b
(a, b)
utils/config_utils.py:784
Method_to_dict
(data)
utils/config_utils.py:830
Method_to_lazy_dict
Convert the ConfigDict to a normal dictionary recursively, and keep the ``LazyObject`` or ``LazyAttr`` object not built.
utils/config_utils.py:826
Method_to_lazy_dict
Convert config object to dictionary with lazy object, and filter the imported object.
utils/config_utils.py:2196
Functionaccumulate_results
(all_results)
video_data_preprocess/create_nuplan_json.py:159
Functionallgather
(t: torch.Tensor, cat=True)
utils/dist.py:113
Functionallgather_diff_shape
(t: torch.Tensor, cat=True)
utils/dist.py:126
Functionapply_rotary_emb
(xq, xk, freqs_cis)
utils/rope_2d.py:31
Methodauto_argparser
Generate argparser from config file automatically (experimental)
utils/config_utils.py:1847
← previousnext →101–200 of 280, ranked by callers