MCPcopy Create free account

hub / github.com/microsoft/Cream / functions

Functions4,396 in github.com/microsoft/Cream

↓ 4 callersMethodset_epoch
r""" Sets the epoch for this sampler. When :attr:`shuffle=True`, this ensures all replicas use a different random ordering for each ep
TinyViT/data/sampler.py:148
↓ 4 callersMethodset_epoch
(self, epoch)
CDARTS/benchmark201/datasets/data_utils.py:71
↓ 4 callersMethodsplit_zip_style_path
(path)
MiniViT/Mini-Swin/data/zipreader.py:32
↓ 4 callersMethodupdate
(self, inter, union, N)
CDARTS/CDARTS_segmentation/train/seg_metrics.py:11
↓ 4 callersFunctionworld_info_from_env
()
TinyCLIP/src/training/distributed.py:43
↓ 4 callersFunctionwrap_fp16_model
(model)
CDARTS/CDARTS_detection/mmdet/core/fp16/hooks.py:85
↓ 4 callersMethodxyxy2xywh
(self, bbox)
CDARTS/CDARTS_detection/mmdet/datasets/coco.py:120
↓ 3 callersMethod__init__
(self, backbone, position_embedding)
iRPE/DETR-with-iRPE/models/backbone.py:97
↓ 3 callersMethod__init__
(self, padding, fill=0, padding_mode='constant')
TinyViT/data/augmentation/aug_tv_transforms.py:271
↓ 3 callersMethod__init__
(self, embed_dim: int, # vision image_resolution: int,
TinyViT/models/clip.py:109
↓ 3 callersMethod__init__
assuming input size 8x8 or 16x16
CDARTS/benchmark201/models/aux_head.py:7
↓ 3 callersMethod__init__
assuming input size 8x8 or 16x16
CDARTS/lib/models/aux_head.py:7
↓ 3 callersMethod__init__
(self, in_channels, out_channels, kernel_size,
CDARTS/CDARTS_detection/mmdet/ops/dcn/modules/deform_conv.py:12
↓ 3 callersMethod__init__
(self, block_args, num_classes=1000, in_chans=3, stem_size=16, num_features=1280, head_bias=True,
CDARTS/CDARTS_detection/mmdet/models/backbones/mobilenetv3.py:142
↓ 3 callersMethod__init__
(self, in_channels, out_channels, expansion, stride, kernel_size, dilation=1, groups=1)
CDARTS/CDARTS_detection/mmdet/models/bbox_heads/convfc_bbox_head.py:48
↓ 3 callersMethod__init__
(self, inplanes, planes, stride=1, downsample=None, groups=1, base_width=64, dilation=1, norm
CDARTS/CDARTS_segmentation/segmentation/model/backbone/hrnet.py:57
↓ 3 callersMethod__init__
(self, gamma=2, alpha=None, ignore_index=-100, reduction='mean')
CDARTS/CDARTS_segmentation/train/loss.py:35
↓ 3 callersMethod__init__
(self, block_args, num_classes=1000, in_chans=3, stem_size=16, num_features=1280, head_bias=True,
CDARTS/CDARTS_segmentation/train/cydas.py:148
↓ 3 callersMethod__init__
(self, dim, input_resolution, depth, num_heads, window_size, mlp_ratio=4., qkv_bias=True, qk_
MiniViT/Mini-Swin/models/swin_transformer_minivit_distill.py:374
↓ 3 callersMethod__init__
(self, dim, input_resolution, depth, num_heads, window_size, mlp_ratio=4., qkv_bias=True, qk_
MiniViT/Mini-Swin/models/swin_transformer_distill.py:255
↓ 3 callersFunction_copy_to_full_weight
(dst, src)
TinyCLIP/src/open_clip/model.py:1308
↓ 3 callersFunction_find_caller
Returns: str: module name of the caller tuple: a hashable key to be used to identify different callers
CDARTS/CDARTS_segmentation/segmentation/utils/logger.py:114
↓ 3 callersMethod_get_files
Gets files for the specified data type and dataset split. Args: data: String, desired data ('image' or 'label'). datas
CDARTS/CDARTS_segmentation/dataloaders/segdatasets/cityscapes.py:88
↓ 3 callersMethod_get_files
Gets files for the specified data type and dataset split. Args: data: String, desired data ('image' or 'label'). datas
CDARTS/CDARTS_segmentation/segmentation/data/datasets/cityscapes.py:88
↓ 3 callersFunction_get_layer_id
(name)
TinyCLIP/src/open_clip/model.py:1367
↓ 3 callersMethod_get_src_permutation_idx
(self, indices)
iRPE/DETR-with-iRPE/models/detr.py:193
↓ 3 callersFunction_get_warmup_factor_at_iter
Return the learning rate warmup factor at a specific iteration. See https://arxiv.org/abs/1706.02677 for more details. Args: meth
CDARTS/CDARTS_segmentation/segmentation/solver/lr_scheduler.py:138
↓ 3 callersFunction_hrnet
(arch, pretrained, progress, **kwargs)
CDARTS/CDARTS_segmentation/segmentation/model/backbone/hrnet.py:489
↓ 3 callersMethod_init_params
(self)
CDARTS/CDARTS_segmentation/segmentation/model/meta_arch/base.py:24
↓ 3 callersFunction_make_divisible
This function is taken from the original tf repo. It ensures that all layers have a channel number that is divisible by 8 It can be seen
CDARTS/CDARTS_segmentation/segmentation/model/backbone/mobilenet.py:18
↓ 3 callersMethod_make_stage
(self, layer_config, in_channels, multiscale_output=True)
CDARTS/CDARTS_detection/mmdet/models/backbones/hrnet.py:393
↓ 3 callersMethod_make_stage
(self, layer_config, num_inchannels, multi_scale_output=True)
CDARTS/CDARTS_segmentation/segmentation/model/backbone/hrnet.py:408
↓ 3 callersMethod_make_transition_layer
(self, num_channels_pre_layer, num_channels_cur_layer)
CDARTS/CDARTS_detection/mmdet/models/backbones/hrnet.py:312
↓ 3 callersMethod_make_transition_layer
( self, num_channels_pre_layer, num_channels_cur_layer)
CDARTS/CDARTS_segmentation/segmentation/model/backbone/hrnet.py:356
↓ 3 callersMethod_params_per_batch
(self)
TinyViT/data/augmentation/mixup.py:145
↓ 3 callersFunction_run
(command)
iRPE/DETR-with-iRPE/util/misc.py:251
↓ 3 callersFunction_warmup_lr
(base_lr, warmup_length, step)
TinyCLIP/src/training/scheduler.py:13
↓ 3 callersMethodadd_architecture_params
(self, n_ops)
CDARTS/lib/models/cdarts_controller.py:406
↓ 3 callersMethodadd_meter
(self, name, meter)
iRPE/DETR-with-iRPE/util/misc.py:190
↓ 3 callersFunctionadd_path
(path)
CDARTS/CDARTS_segmentation/train/_init_paths.py:10
↓ 3 callersMethodafter_iter
(self, runner)
CDARTS/CDARTS_detection/mmcv/runner/hooks/hook.py:18
↓ 3 callersFunctionallreduce_grads
(params, coalesce=True, bucket_size_mb=-1)
CDARTS/CDARTS_detection/mmdet/core/utils/dist_utils.py:32
↓ 3 callersFunctionanchor_target
Compute regression and classification targets for anchors. Args: anchor_list (list[list]): Multi level anchors of each image. val
CDARTS/CDARTS_detection/mmdet/core/anchor/anchor_target.py:7
↓ 3 callersMethodas_markdown
Return configs as markdown format
CDARTS/lib/config.py:31
↓ 3 callersMethodbackward
- Inputs grad_output: float32 (B, H, L_query, L_key) - Outputs grad_input: float32 (B, H, L_query, nu
iRPE/DETR-with-iRPE/rpe_ops/rpe_index.py:42
↓ 3 callersFunctionbbox_mapping
Map bboxes from the original image scale to testing scale
CDARTS/CDARTS_detection/mmdet/core/bbox/transforms.py:91
↓ 3 callersMethodboard_size
(self)
Cream/lib/models/PrioritizedBoard.py:45
↓ 3 callersMethodbuild_hook
(self, args, hook_type=None)
CDARTS/CDARTS_detection/mmcv/runner/runner.py:226
↓ 3 callersFunctionbuild_search_head
Build head model from config dict.
CDARTS/CDARTS_detection/mmdet/models/bbox_heads/auto_head/build_head.py:9
↓ 3 callersFunctionbuild_transform
(is_train, args)
iRPE/DeiT-with-iRPE/datasets.py:78
↓ 3 callersFunctionbuild_transform
(is_train, args)
MiniViT/Mini-DeiT/datasets.py:79
↓ 3 callersFunctionbuild_transforms
(dataset, is_train=True)
CDARTS/CDARTS_segmentation/dataloaders/transforms/build.py:9
↓ 3 callersFunctionbuild_transforms
(dataset, is_train=True)
CDARTS/CDARTS_segmentation/segmentation/data/transforms/build.py:9
↓ 3 callersFunctioncalc_dropout
(dropout, sample_embed_dim, super_embed_dim)
AutoFormer/model/supernet_transformer.py:306
↓ 3 callersMethodcalculate_model_size
(self, zs)
TinyCLIP/src/open_clip/l0module.py:266
↓ 3 callersMethodclear
(self)
CDARTS/CDARTS_detection/mmcv/runner/log_buffer.py:14
↓ 3 callersFunctionconv1x1
1x1 convolution
CDARTS/CDARTS_segmentation/segmentation/model/backbone/resnet.py:34
↓ 3 callersFunctionconv1x1_dwconv_conv1x1
(prefix, in_channels, out_channels, mid_channels, kernel_size, stride, bn_training=True)
CDARTS/CDARTS_detection/mmdet/models/backbones/detnas.py:83
↓ 3 callersFunctionconv3x3
3x3 convolution with padding
CDARTS/CDARTS_segmentation/segmentation/model/backbone/hrnet.py:43
↓ 3 callersFunctionconv3x3
3x3 convolution with padding
CDARTS/CDARTS_segmentation/segmentation/model/backbone/resnet.py:28
↓ 3 callersFunctionconvert_bn
(blobs, state_dict, caffe_name, torch_name, converted_names)
CDARTS/CDARTS_detection/tools/detectron2pytorch.py:10
↓ 3 callersFunctionconvert_conv_fc
(blobs, state_dict, caffe_name, torch_name, converted_names)
CDARTS/CDARTS_detection/tools/detectron2pytorch.py:23
↓ 3 callersFunctionconvert_to_new_checkpoint
(state_dict, used_ddp=False)
TinyCLIP/src/open_clip/model.py:1115
↓ 3 callersFunctionconvert_video
Convert a video with ffmpeg. This provides a general api to ffmpeg, the executed command is:: `ffmpeg -y <pre_options> -i <in_file> <opt
CDARTS/CDARTS_detection/mmcv/video/processing.py:10
↓ 3 callersFunctioncreate_conv2d
Select a 2d convolution implementation based on arguments Creates and returns one of torch.nn.Conv2d, Conv2dSame, MixedConv2d, or CondConv2d.
CDARTS/CDARTS_detection/mmdet/models/backbones/efficientnet.py:141
↓ 3 callersFunctioncreate_conv2d_pad
(in_chs, out_chs, kernel_size, **kwargs)
CDARTS/CDARTS_detection/mmdet/models/backbones/builder.py:358
↓ 3 callersFunctioncreate_exp_dir
(path, scripts_to_save=None)
CDARTS/CDARTS_segmentation/tools/utils/darts_utils.py:85
↓ 3 callersFunctioncrop
(image, target, region)
iRPE/DETR-with-iRPE/datasets/transforms.py:16
↓ 3 callersMethodddpify
(self, ddp_fn)
TinyCLIP/src/open_clip/model.py:977
↓ 3 callersMethoddecompose
(self)
iRPE/DETR-with-iRPE/util/misc.py:299
↓ 3 callersFunctiondistance2bbox
Decode distance prediction to bounding box. Args: points (Tensor): Shape (n, 2), [x, y]. distance (Tensor): Distance from the giv
CDARTS/CDARTS_detection/mmdet/core/bbox/transforms.py:159
↓ 3 callersFunctiondrop_path
Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks). This is the same as the DropConnect impl I created for E
AutoFormer/model/utils.py:71
↓ 3 callersFunctionefficientnet_init_weights
(model: nn.Module, init_fn=None, zero_gamma=False)
CDARTS/CDARTS_detection/mmdet/models/backbones/builder.py:859
↓ 3 callersFunctioneval_map
Evaluate mAP of a dataset. Args: det_results (list[list]): [[cls1_det, cls2_det, ...], ...]. The outer list indicates images,
CDARTS/CDARTS_detection/mmdet/core/evaluation/mean_ap.py:266
↓ 3 callersFunctionevaluate
(model, data, epoch, args, tb_writer=None, step=None, num_feed_images=None)
TinyCLIP/src/training/train.py:708
↓ 3 callersFunctionexpand_urls
(urls, weights=None)
TinyCLIP/src/training/data.py:80
↓ 3 callersMethodextract_feat
(self, img)
CDARTS/CDARTS_detection/mmdet/models/detectors/two_stage.py:88
↓ 3 callersMethodextract_feat
(self, img)
CDARTS/CDARTS_detection/mmdet/models/detectors/single_stage.py:39
↓ 3 callersMethodfeature_channels
Feature Channel Shortcut Returns feature channel count for each output index if idx == None. If idx is an integer, will return featur
CDARTS/CDARTS_detection/mmdet/models/backbones/efficientnet.py:593
↓ 3 callersMethodflops
(self)
MiniViT/Mini-Swin/models/swin_transformer_minivit_distill.py:439
↓ 3 callersMethodflops
(self)
MiniViT/Mini-Swin/models/swin_mlp.py:454
↓ 3 callersMethodflops
(self)
MiniViT/Mini-Swin/models/swin_transformer.py:394
↓ 3 callersMethodflops
(self)
MiniViT/Mini-Swin/models/swin_transformer_minivit.py:529
↓ 3 callersMethodflops
(self)
MiniViT/Mini-Swin/models/swin_transformer_distill.py:300
↓ 3 callersMethodforward
(ctx, input)
CDARTS/CDARTS_segmentation/train/layers.py:34
↓ 3 callersMethodgenerate_nas_layer
(self, C_cur, C_p, C_pp, reduction_p, reduction_cur, genotype, cell_num=3, bn_affine=True)
CDARTS/lib/models/cdarts_controller.py:456
↓ 3 callersMethodget_class_colors
(*args)
CDARTS/CDARTS_segmentation/tools/datasets/bdd/bdd.py:10
↓ 3 callersFunctionget_classes
Get class names of a dataset.
CDARTS/CDARTS_detection/mmdet/core/evaluation/class_names.py:94
↓ 3 callersFunctionget_condconv_initializer
(initializer, num_experts, expert_shape)
CDARTS/CDARTS_detection/mmdet/models/backbones/efficientnet_builder.py:181
↓ 3 callersFunctionget_condconv_initializer
(initializer, num_experts, expert_shape)
CDARTS/CDARTS_detection/mmdet/models/backbones/builder.py:181
↓ 3 callersFunctionget_config
Get a yacs CfgNode object with default values.
TinyViT/config.py:244
↓ 3 callersFunctionget_logger
Make python logger
Cream/lib/utils/util.py:53
↓ 3 callersMethodget_num_parameters_and_constraint
(self, hidden=False)
TinyCLIP/src/open_clip/l0module.py:150
↓ 3 callersMethodget_sampled_params_numel
(self, config)
AutoFormer/model/supernet_transformer.py:129
↓ 3 callersFunctionget_shared_folder
()
iRPE/DeiT-with-iRPE/run_with_submitit.py:30
↓ 3 callersFunctionget_shared_folder
()
iRPE/DETR-with-iRPE/run_with_submitit.py:24
↓ 3 callersFunctionget_single_rpe_config
Get the config of single relative position encoding Parameters ---------- ratio: float The ratio to control the number of buckets
iRPE/DeiT-with-iRPE/irpe.py:770
↓ 3 callersFunctionget_single_rpe_config
Get the config of single relative position encoding Parameters ---------- ratio: float The ratio to control the number of buckets
iRPE/DETR-with-iRPE/models/rpe_attention/irpe.py:770
↓ 3 callersFunctionget_single_rpe_config
Get the config of single relative position encoding Parameters ---------- ratio: float The ratio to control the number of buckets
MiniViT/Mini-DeiT/irpe.py:770
↓ 3 callersMethodget_target_sparsity
(self, pruned_steps)
TinyCLIP/src/open_clip/l0module.py:204
← previousnext →301–400 of 4,396, ranked by callers