MCPcopy Create free account

hub / github.com/amazon-science/mm-cot / functions

Functions1,823 in github.com/amazon-science/mm-cot

↓ 1 callersFunction_gen_mnasnet_small
Creates a mnasnet-b1 model. Ref impl: https://github.com/tensorflow/tpu/tree/master/models/official/mnasnet Paper: https://arxiv.org/pdf/1807
timm/models/efficientnet.py:634
↓ 1 callersFunction_gen_mobilenet_v3_rw
Creates a MobileNet-V3 model. Ref impl: ? Paper: https://arxiv.org/abs/1905.02244 Args: channel_multiplier: multiplier to number o
timm/models/mobilenetv3.py:239
↓ 1 callersFunction_gen_spnasnet
Creates the Single-Path NAS model from search targeted for Pixel1 phone. Paper: https://arxiv.org/abs/1904.02877 Args: channel_multipl
timm/models/efficientnet.py:723
↓ 1 callersMethod_get_lr
(param_group, param_state)
timm/optim/adafactor.py:56
↓ 1 callersMethod_get_options
(param_group, param_shape)
timm/optim/adafactor.py:67
↓ 1 callersFunction_get_pixels
(per_pixel, rand_color, patch_size, dtype=torch.float32, device='cuda')
timm/data/random_erasing.py:13
↓ 1 callersFunction_get_return_layers
(feature_info, out_map)
timm/models/features.py:145
↓ 1 callersMethod_get_stage_params
(self, cfg, default_stride=2, output_stride=32, drop_path_rate=0.)
timm/models/regnet.py:284
↓ 1 callersFunction_init_weights
(m, n='')
timm/models/byobnet.py:1134
↓ 1 callersMethod_initialize_weights
(self)
timm/models/vgg.py:148
↓ 1 callersFunction_interpolation
(kwargs)
timm/data/auto_augment.py:44
↓ 1 callersFunction_label_from_paths
(*path, leaf_only=True)
timm/data/parsers/parser_image_in_tar.py:118
↓ 1 callersMethod_load_checkpoint
(self, checkpoint_path)
timm/utils/model_ema.py:51
↓ 1 callersFunction_load_weights
(model: nn.Module, checkpoint_path: str, prefix: str = 'resnet/')
timm/models/resnetv2.py:430
↓ 1 callersFunction_load_weights
Load weights from .npz checkpoints for official Google Brain Flax implementation
timm/models/vision_transformer.py:399
↓ 1 callersMethod_make_block
(self, ba, block_idx, block_count)
timm/models/efficientnet_builder.py:298
↓ 1 callersMethod_make_branches
(self, num_branches, block, num_blocks, num_channels)
timm/models/hrnet.py:435
↓ 1 callersMethod_make_fuse_layers
(self)
timm/models/hrnet.py:442
↓ 1 callersMethod_make_one_branch
(self, branch_index, block, num_blocks, num_channels, stride=1)
timm/models/hrnet.py:418
↓ 1 callersMethod_mix_batch
(self, x)
timm/data/mixup.py:196
↓ 1 callersMethod_mix_batch_collate
(self, output, batch)
timm/data/mixup.py:280
↓ 1 callersMethod_mix_elem
(self, x)
timm/data/mixup.py:159
↓ 1 callersMethod_mix_elem_collate
(self, output, batch, half=False)
timm/data/mixup.py:227
↓ 1 callersMethod_mix_pair
(self, x)
timm/data/mixup.py:176
↓ 1 callersMethod_mix_pair_collate
(self, output, batch)
timm/data/mixup.py:252
↓ 1 callersFunction_posterize_level_to_arg
(level, _hparams)
timm/data/auto_augment.py:222
↓ 1 callersMethod_projection
(self, p, grad, perturb, delta, wd_ratio, eps)
timm/optim/adamp.py:39
↓ 1 callersMethod_projection
(self, p, grad, perturb, delta, wd_ratio, eps)
timm/optim/sgdp.py:39
↓ 1 callersFunction_search_split
(root, split)
timm/data/dataset_factory.py:6
↓ 1 callersFunction_select_rand_weights
(weight_idx=0, transforms=None)
timm/data/auto_augment.py:604
↓ 1 callersFunction_solarize_level_to_arg
(level, _hparams)
timm/data/auto_augment.py:243
↓ 1 callersFunction_weight_init
(m)
timm/models/senet.py:62
↓ 1 callersFunction_xception
(variant, pretrained=False, **kwargs)
timm/models/xception.py:222
↓ 1 callersFunctionact_with_gamma
(act_type, gamma: float = 1.)
timm/models/nfnet.py:298
↓ 1 callersFunctionadapt_model_from_file
(parent_module, model_variant)
timm/models/helpers.py:322
↓ 1 callersFunctionadapt_model_from_string
(parent_module, model_string)
timm/models/helpers.py:272
↓ 1 callersFunctionadaptive_clip_grad
(parameters, clip_factor=0.01, eps=1e-3, norm_type=2.0)
timm/utils/agc.py:30
↓ 1 callersFunctionadd_weight_decay
(model, weight_decay=1e-5, skip_list=())
timm/optim/optim_factory.py:29
↓ 1 callersFunctionadjust_widths_groups_comp
Adjusts the compatibility of widths and groups.
timm/models/regnet.py:107
↓ 1 callersMethodany_requires_grad
(self, x)
timm/models/densenet.py:67
↓ 1 callersFunctionaugment_and_mix_transform
Create AugMix PyTorch transform :param config_str: String defining configuration of random augmentation. Consists of multiple sections separated
timm/data/auto_augment.py:775
↓ 1 callersFunctionaugmix_ops
(magnitude=10, hparams=None, transforms=None)
timm/data/auto_augment.py:702
↓ 1 callersFunctionauto_augment_policy
(name='v0', hparams=None)
timm/data/auto_augment.py:485
↓ 1 callersFunctionauto_augment_policy_original
(hparams)
timm/data/auto_augment.py:419
↓ 1 callersFunctionauto_augment_policy_originalr
(hparams)
timm/data/auto_augment.py:452
↓ 1 callersFunctionauto_augment_policy_v0
(hparams)
timm/data/auto_augment.py:352
↓ 1 callersFunctionauto_augment_policy_v0r
(hparams)
timm/data/auto_augment.py:385
↓ 1 callersFunctionauto_augment_transform
Create a AutoAugment transform :param config_str: String defining configuration of auto augmentation. Consists of multiple sections separate
timm/data/auto_augment.py:511
↓ 1 callersFunctionbleu_score
(reference, hypothesis, gram)
evaluations.py:19
↓ 1 callersFunctioncaculate_rouge
(results, data)
evaluations.py:61
↓ 1 callersFunctioncaculate_similariry
(results, data, model)
evaluations.py:89
↓ 1 callersMethodcall_checkpoint_bottleneck
(self, x)
timm/models/densenet.py:75
↓ 1 callersMethodchannels
feature channels accessor
timm/models/features.py:62
↓ 1 callersFunctioncreate_block
(block: Union[str, nn.Module], **kwargs)
timm/models/byobnet.py:860
↓ 1 callersFunctioncreate_byob_stages
( cfg: ByoModelCfg, drop_path_rate: float, output_stride: int, stem_feat: Dict[str, Any], feat
timm/models/byobnet.py:999
↓ 1 callersFunctioncreate_byob_stem
(in_chs, out_chs, stem_type='', pool_type='', feat_prefix='stem', layers: LayerFn = None)
timm/models/byobnet.py:914
↓ 1 callersFunctioncreate_optimizer_v2
Create an optimizer. TODO currently the model is passed in and all parameters are selected for optimization. For more general use an interfa
timm/optim/optim_factory.py:73
↓ 1 callersFunctioncreate_stem
( in_chans=3, out_chs=32, kernel_size=3, stride=2, pool='', act_layer=None, norm_layer=None, a
timm/models/cspnet.py:131
↓ 1 callersFunctioncreate_stem
(in_chs, out_chs, stem_type='', conv_layer=None, act_layer=None, preact_feature=True)
timm/models/nfnet.py:396
↓ 1 callersFunctiondistorted_bounding_box_crop
Generates cropped_image using one of the bboxes randomly distorted. See `tf.image.sample_distorted_bounding_box` for more documentation. Arg
timm/data/tf_preprocessing.py:32
↓ 1 callersFunctiondownload_cached_file
(url, check_hash=True, progress=False)
timm/models/hub.py:41
↓ 1 callersFunctiondownsample_avg
( in_channels, out_channels, kernel_size, stride=1, dilation=1, first_dilation=None, norm_layer=None)
timm/models/resnet.py:440
↓ 1 callersFunctiondownsample_conv
( in_channels, out_channels, kernel_size, stride=1, dilation=1, first_dilation=None, norm_layer=None)
timm/models/resnet.py:426
↓ 1 callersFunctiondownsample_conv
( in_chs, out_chs, kernel_size, stride=1, dilation=1, norm_layer=None)
timm/models/regnet.py:177
↓ 1 callersFunctiondrop_blocks
(drop_block_rate=0.)
timm/models/resnet.py:457
↓ 1 callersFunctionefficientnet_b2
EfficientNet-B2
timm/models/efficientnet.py:1262
↓ 1 callersFunctionefficientnet_b3
EfficientNet-B3
timm/models/efficientnet.py:1278
↓ 1 callersFunctioneven_split_indices
(split, n, num_samples)
timm/data/parsers/parser_tfds.py:32
↓ 1 callersFunctionexpand_blocks_cfg
(stage_blocks_cfg: Union[ByoBlockCfg, Sequence[ByoBlockCfg]])
timm/models/byobnet.py:480
↓ 1 callersFunctionextract_features
(img_type, input_image)
extract_features.py:20
↓ 1 callersFunctionextract_layer
(model, layer)
timm/models/helpers.py:230
↓ 1 callersFunctionextract_tarinfo
(tarfile, class_to_idx=None, sort=True)
timm/data/parsers/parser_image_tar.py:17
↓ 1 callersFunctionextract_tarinfos
(root, class_name_to_idx=None, cache_tarinfo=None, extensions=IMG_EXTENSIONS, sort=True)
timm/data/parsers/parser_image_in_tar.py:63
↓ 1 callersMethodfeat_channels
(self, exp=False)
timm/models/rexnet.py:80
↓ 1 callersMethodfeature_info
(self, location)
timm/models/efficientnet_blocks.py:63
↓ 1 callersMethodfilename
(self, index, basename=False, absolute=False)
timm/data/dataset.py:58
↓ 1 callersMethodfilenames
(self, basename=False, absolute=False)
timm/data/dataset.py:61
↓ 1 callersFunctionfilter_kwargs
(kwargs, names)
timm/models/helpers.py:367
↓ 1 callersFunctionfind_images_and_targets
(folder, types=IMG_EXTENSIONS, class_to_idx=None, leaf_name_only=True, sort=True)
timm/data/parsers/parser_image_folder.py:17
↓ 1 callersMethodforward_features
(self, x)
timm/models/levit.py:517
↓ 1 callersMethodforward_features
(self, x)
timm/models/pit.py:222
↓ 1 callersMethodforward_features
(self, x)
timm/models/xception.py:182
↓ 1 callersMethodforward_features
(self, x)
timm/models/cait.py:280
↓ 1 callersMethodforward_features
(self, x)
timm/models/vovnet.py:330
↓ 1 callersMethodforward_features
(self, x)
timm/models/pnasnet.py:307
↓ 1 callersMethodforward_features
(self, x)
timm/models/gluon_xception.py:188
↓ 1 callersMethodforward_features
(self, x)
timm/models/ghostnet.py:197
↓ 1 callersMethodforward_features
(self, x)
timm/models/resnetv2.py:402
↓ 1 callersMethodforward_features
(self, x)
timm/models/inception_v3.py:382
↓ 1 callersMethodforward_features
(self, x)
timm/models/inception_v3.py:411
↓ 1 callersMethodforward_features
(self, x)
timm/models/xception_aligned.py:163
↓ 1 callersMethodforward_features
(self, x)
timm/models/senet.py:377
↓ 1 callersMethodforward_features
(self, x)
timm/models/hrnet.py:704
↓ 1 callersMethodforward_features
(self, x)
timm/models/dpn.py:251
↓ 1 callersMethodforward_features
(self, x)
timm/models/densenet.py:260
↓ 1 callersMethodforward_features
(self, x)
timm/models/byobnet.py:1122
↓ 1 callersMethodforward_features
(self, x)
timm/models/mobilenetv3.py:144
↓ 1 callersMethodforward_features
(self, x)
timm/models/visformer.py:275
↓ 1 callersMethodforward_features
(self, x)
timm/models/rexnet.py:173
↓ 1 callersMethodforward_features
(self, x)
timm/models/tresnet.py:241
← previousnext →301–400 of 1,823, ranked by callers