MCPcopy Create free account

hub / github.com/YvanYin/DrivingWorld / functions

Functions280 in github.com/YvanYin/DrivingWorld

Functionbroadcast
(t: torch.Tensor, src_rank)
utils/dist.py:153
Methodbuild
Return imported object. Returns: Any: Imported object
utils/config_utils.py:495
Functioncreate_gif
(args, imgs, video_save_path, border_size = 10, fps=2)
utils/testing_utils.py:47
Functioncreate_mp4
(args, imgs, video_save_path, border_size = 10, fps=2)
utils/testing_utils.py:27
Functioncreate_video
(args, imgs, video_save_path, border_size = 10)
utils/testing_utils.py:6
Functiondata_aug_for_seq
(imgs, img_h, img_w)
datasets/datasets_utils.py:14
Methoddecode_code
(self, code_b, shape=None, channel_first=True)
modules/tokenizers/vq_model.py:88
Methoddiff
(cfg1: Union[str, 'Config'], cfg2: Union[str, 'Config'])
utils/config_utils.py:2127
Functiondist_fmt_vals
(val: float, fmt: Union[str, None] = '%.2f')
utils/dist.py:163
Methodencode_hacked
(self, x)
modules/tokenizers/vq_model.py:73
Methodenv_variables
get used environment variables.
utils/config_utils.py:1869
Methodfilename
get file name of config.
utils/config_utils.py:1859
Functionfinalize
()
utils/dist.py:213
Functionflow2rgb
(flow_map_np)
utils/utils.py:24
Functionfor_visualize
(func)
utils/dist.py:201
Methodforward
(self, input)
modules/tokenizers/vq_model.py:107
Methodforward
(self, input)
modules/tokenizers/vq_model.py:146
Methodforward
(self, x)
modules/tokenizers/vq_model.py:191
Methodforward
(self, z)
modules/tokenizers/vq_model.py:242
Methodforward
(self, z)
modules/tokenizers/vq_model.py:275
Methodforward
(self, x)
modules/tokenizers/vq_model.py:377
Methodforward
(self, x)
modules/tokenizers/vq_model.py:405
Methodforward
( self, x: torch.Tensor, freqs_cis: torch.Tensor, mask: Optional[torch.Tensor]
modules/tokenizers/vq_model.py:448
Methodforward
( self, x: torch.Tensor, freqs_cis: torch.Tensor, mask: Optional[torch.Tensor]
modules/tokenizers/vq_model.py:488
Methodforward
(self, x)
modules/tokenizers/vq_model.py:553
Methodforward
(self, x)
modules/tokenizers/vq_model.py:565
Methodforward
(self, x)
modules/tokenizers/vq_model.py:623
Methodforward
(self, x)
modules/tokenizers/vq_model.py:638
Methodforward
(self, x, attn_mask)
models/ar.py:46
Methodforward
(self, x, attn_mask)
models/ar.py:80
Methodforward
(self, x,attn_mask)
models/ar.py:112
Methodforward
(self, x, attn_mask)
models/ar.py:146
Methodforward
(self, x, attn_mask)
models/ar.py:171
Methodforward
(self, x, attn_mask)
models/ar.py:197
Methodforward
(self, x, time_attn_mask,space_attn_mask)
models/ar.py:209
Methodforward
(self, feature_total, pose_indices_total, yaw_indices_total, drop_flag=False)
models/ar.py:381
Methodforward
(self, token, feature, pose, yaw, token_gt, feature_gt, pose_gt, yaw_gt, vqvae_codebook=None, eval=False)
models/model.py:164
Methodfromstring
Build a Config instance from config text. Args: cfg_str (str): Config text. file_format (str): Config file format cor
utils/config_utils.py:1007
Methodget_block_size
(self)
models/ar.py:331
Functionget_deepspeed_config
(args)
utils/deepspeed_utils.py:3
Functionget_device
()
utils/dist.py:64
Functionget_timestep_embedding
This matches the implementation in Denoising Diffusion Probabilistic Models: Create sinusoidal timestep embeddings. Args timesteps (
utils/embeddings.py:3
Functionget_undistort_map
Rectify camera distortions. Camera intrinsic parameters and distortions examples are as follows. camera_matrix = np.array([[fx, 0, cx
utils/undistort_image.py:5
Functionget_world_size
()
utils/dist.py:60
Functionindices_to_pose
(idx_x, idx_y, x_division=128, y_divisions=128)
modules/tokenizers/pose_tokenizer.py:21
Functionindices_to_yaws
(idx_yaw, division=512)
modules/tokenizers/pose_tokenizer.py:40
Functioninit_lr_schedule
(optimizer, milstones=[1000000, 1500000, 2000000], gamma=0.5)
utils/running.py:15
Functioninit_optimizer
(model, lr=1e-4, weight_decay=1e-3)
utils/running.py:10
Functioninitialize
(fork=False, backend='nccl', gpu_id_if_not_distibuted=0, timeout=30)
utils/dist.py:20
Methodis_base_line
(c)
utils/config_utils.py:1719
Methodlast_layer
(self)
modules/tokenizers/vq_model.py:239
Functionload_from_deepspeed_ckpt
(args, model)
utils/running.py:150
Functionlocal_master_only
(func)
utils/dist.py:188
Functionlog_dict_args
Logs a dictionary of arguments. Args: dict_args (dict): The dictionary of arguments to be logged.
utils/running.py:176
Functionmaster_only
(func)
utils/dist.py:175
Methodmerge
Merge another dictionary into current dictionary. Args: other (dict): Another dictionary.
utils/config_utils.py:776
Methodmodule
(self)
utils/config_utils.py:118
Methodmodule
(self)
utils/config_utils.py:532
Functionnew_group
(ranks: List[int])
utils/dist.py:90
Methodpretty_text
get formatted python config text.
utils/config_utils.py:1874
Functionresume_ckpt
(local_rank, args, model, optimizer=None)
utils/running.py:71
Functionreverse_seq_data
(poses, seqs)
datasets/datasets_utils.py:6
Functionrgb2ycbcr
(img_np)
utils/utils.py:34
Functionsave_ckpt
Save the model, optimizer, lr scheduler.
utils/running.py:54
Functionsave_ckpt_deepspeed
Save the model, optimizer, lr scheduler.
utils/running.py:134
Functionsave_model
(self, path, epoch, rank=0)
utils/running.py:20
Methodsave_model
(self, path, epoch, rank=0)
models/model.py:212
Functionset_gpu_id
(gpu_id: int)
utils/dist.py:68
Functionsetup_logger
(logger_name, save_dir, phase, level=logging.INFO, screen=False, to_file=False)
utils/utils.py:9
Methodtext
get config text.
utils/config_utils.py:1864
Methodtokenize_pose
(self, poses)
modules/tokenizers/model_tokenizer.py:71
Methodtop_k_logits
(self, logits, k)
models/model.py:172
Functionundistort_image
(img, map1, map2, roi)
utils/undistort_image.py:38
Methodvalues
Yield the values of the dictionary. If class attribute ``lazy`` is False, the value of ``LazyObject`` or ``LazyAttr`` will be built a
utils/config_utils.py:754
Methodvisit_Assign
(self, node)
utils/config_utils.py:41
Methodvisit_Import
Work with ``_gather_abs_import_lazyobj`` to hack the ``import ...`` syntax. Examples: >>> import mmcls.models
utils/config_utils.py:393
Methodvisit_ImportFrom
Hack the ``from ... import ...`` syntax and update the global_dict. Examples: >>> from mmdet.models import RetinaNet Wil
utils/config_utils.py:299
Functionwrapper
(*args, **kwargs)
utils/dist.py:177
Methodz_to_clips_all
(self, indices)
modules/tokenizers/model_tokenizer.py:51
Methodz_to_image
(self, indices)
modules/tokenizers/model_tokenizer.py:35
← previous201–280 of 280, ranked by callers