MCPcopy Create free account

hub / github.com/KaiyangZhou/Dassl.pytorch / functions

Functions664 in github.com/KaiyangZhou/Dassl.pytorch

⨍ Functions 664◇ Types & classes 136
↓ 2 callersMethodload_model
(self, directory, epoch=None)
dassl/engine/trainer.py:172
↓ 2 callersFunctionmixup
Mixup. Args: x1 (torch.Tensor): data with shape of (b, c, h, w). x2 (torch.Tensor): data with shape of (b, c, h, w). y1 (
dassl/modeling/ops/mixup.py:4
↓ 2 callersFunctionmkdir_if_missing
(directory)
datasets/da/digit5.py:9
↓ 2 callersMethodpairwise_distance
(self, x, u)
dassl/engine/da/m3sda.py:166
↓ 2 callersFunctionparse_function
(*metrics, directory="", args=None, end_signal=None)
tools/parse_test_res.py:64
↓ 2 callersMethodsave_model
( self, epoch, directory, is_best=False, val_result=None, model_name="" )
dassl/engine/trainer.py:118
↓ 2 callersMethodset_activation_status
(self, status=True)
dassl/modeling/ops/efdmix.py:81
↓ 2 callersMethodset_activation_status
(self, status=True)
dassl/modeling/ops/mixstyle.py:81
↓ 2 callersFunctionsigmoid_rampup
Exponential rampup. Args: current (int): current step. rampup_length (int): maximum step.
dassl/modeling/ops/utils.py:43
↓ 2 callersMethodstep
(self, closure=None)
dassl/optim/radam.py:50
↓ 2 callersMethodtest
A generic testing pipeline.
dassl/engine/trainer.py:446
↓ 2 callersFunctionupdate_file
(filename, text_to_search, replacement_text)
tools/replace_text.py:17
↓ 2 callersMethodupdate_mix_method
(self, mix="random")
dassl/modeling/ops/efdmix.py:84
↓ 2 callersMethodupdate_mix_method
(self, mix="random")
dassl/modeling/ops/mixstyle.py:84
↓ 1 callersFunctionCutoutAbs
(img, v)
dassl/data/transforms/randaugment.py:137
↓ 1 callersMethod__init__
( self, dist_metric="cosine", eps=0.01, max_iter=5, bp_to_sinkhorn=Fal
dassl/modeling/ops/optimal_transport.py:38
↓ 1 callersMethod__init__
(self, c_hidden=64)
dassl/modeling/backbone/cnn_digitsdg.py:23
↓ 1 callersMethod__init__
(self, block_args, global_params, image_size=None)
dassl/modeling/backbone/efficientnet/model.py:26
↓ 1 callersMethod__init__
(self, cfg)
dassl/data/datasets/dg/cifar_c.py:49
↓ 1 callersMethod__init__
(self, cfg)
dassl/data/datasets/ssl/cifar.py:22
↓ 1 callersMethod__init__
(self, cfg, model_cfg, num_classes, **kwargs)
dassl/engine/trainer.py:27
↓ 1 callersMethod__init__
(self, cfg)
dassl/engine/dg/daeldg.py:38
↓ 1 callersMethod__init__
(self, cfg)
dassl/engine/da/m3sda.py:33
↓ 1 callersMethod__init__
(self, cfg)
dassl/engine/da/dael.py:36
↓ 1 callersMethod__init__
(self, cfg)
dassl/engine/da/cdac.py:65
↓ 1 callersMethod__init__
(self, cfg)
dassl/engine/da/mme.py:34
↓ 1 callersMethod__init__
(self, cfg)
dassl/evaluation/evaluator.py:13
↓ 1 callersMethod__iter__
(self)
dassl/data/samplers.py:38
↓ 1 callersMethod__iter__
(self)
dassl/data/samplers.py:93
↓ 1 callersMethod__iter__
(self)
dassl/data/samplers.py:149
↓ 1 callersFunction_build_transform_test
(cfg, choices, target_size, normalize)
dassl/data/transforms/transforms.py:326
↓ 1 callersFunction_build_transform_train
(cfg, choices, target_size, normalize)
dassl/data/transforms/transforms.py:206
↓ 1 callersMethod_change_in_channels
(model, in_channels)
dassl/modeling/backbone/efficientnet/model.py:316
↓ 1 callersMethod_check_input
(self, x)
dassl/modeling/ops/transnorm.py:46
↓ 1 callersMethod_check_input
(self, x)
dassl/modeling/backbone/cnn_digitsdg.py:32
↓ 1 callersMethod_check_input
(self, x)
dassl/modeling/backbone/cnn_digit5_m3sda.py:30
↓ 1 callersMethod_check_input
(self, x)
dassl/modeling/backbone/cnn_digitsingle.py:25
↓ 1 callersMethod_decode_block_string
Gets a block through a string notation of arguments.
dassl/modeling/backbone/efficientnet/utils.py:288
↓ 1 callersMethod_encode_block_string
Encodes a block to a string.
dassl/modeling/backbone/efficientnet/utils.py:317
↓ 1 callersMethod_forward_impl
(self, x: Tensor)
dassl/modeling/backbone/resnet_dynamic.py:523
↓ 1 callersMethod_init_params
(self)
dassl/modeling/backbone/resnet.py:171
↓ 1 callersMethod_initialize_weights
(self)
dassl/modeling/backbone/vgg.py:51
↓ 1 callersMethod_make_layer
( self, block, in_planes, out_planes, nb_layers, stride, dropRate )
dassl/modeling/backbone/wide_resnet.py:71
↓ 1 callersMethod_read_data_test
(self, data_dir)
dassl/data/datasets/ssl/cifar.py:79
↓ 1 callersMethod_read_data_train
(self, data_dir, num_labeled, val_percent)
dassl/data/datasets/ssl/cifar.py:43
↓ 1 callersMethod_read_data_train
(self, data_dir, fold, fold_file)
dassl/data/datasets/ssl/stl10.py:52
↓ 1 callersFunction_vgg
(arch, cfg, batch_norm, pretrained)
dassl/modeling/backbone/vgg.py:133
↓ 1 callersMethodafter_epoch
(self)
dassl/engine/trainer.py:263
↓ 1 callersMethodafter_train
(self)
dassl/engine/trainer.py:257
↓ 1 callersMethodassess_y_pred_quality
(self, y_pred, y_true, mask)
dassl/engine/da/cdac.py:139
↓ 1 callersMethodassess_y_pred_quality
(self, y_pred, y_true, mask)
dassl/engine/ssl/fixmatch.py:40
↓ 1 callersMethodbefore_epoch
(self)
dassl/engine/trainer.py:260
↓ 1 callersMethodbefore_train
(self)
dassl/engine/trainer.py:254
↓ 1 callersFunctionbuild_backbone
(name, verbose=True, **kwargs)
dassl/modeling/backbone/build.py:6
↓ 1 callersMethodbuild_conv_block
( self, dim, padding_type, norm_layer, use_dropout, use_bias )
dassl/modeling/network/ddaig_fcn.py:69
↓ 1 callersMethodbuild_critic
(self)
dassl/engine/da/adda.py:38
↓ 1 callersMethodbuild_critic
(self)
dassl/engine/da/dann.py:26
↓ 1 callersMethodbuild_data_loader
Create essential data-related attributes. A re-implementation of this method must create the same attributes (self.dm is optional).
dassl/engine/trainer.py:341
↓ 1 callersFunctionbuild_dataset
(cfg)
dassl/data/datasets/build.py:6
↓ 1 callersFunctionbuild_evaluator
(cfg, **kwargs)
dassl/evaluation/build.py:6
↓ 1 callersMethodbuild_model
Build and register model. The default builds a classification model along with its optimizer and scheduler. Custom trainers
dassl/engine/trainer.py:360
↓ 1 callersFunctionbuild_network
(name, verbose=True, **kwargs)
dassl/modeling/network/build.py:6
↓ 1 callersFunctionbuild_sampler
( sampler_type, cfg=None, data_source=None, batch_size=32, n_domain=0, n_ins=16 )
dassl/data/samplers.py:181
↓ 1 callersFunctionbuild_trainer
(cfg)
dassl/engine/build.py:6
↓ 1 callersMethodcheck_cfg
Check whether some variables are set correctly for the trainer (optional). For example, a trainer might require a particular sampler
dassl/engine/trainer.py:329
↓ 1 callersFunctionclean_cfg
Remove unused trainers (configs). Aim: Only show relevant information when calling print(cfg). Args: cfg (_C): cfg instance.
dassl/config/__init__.py:8
↓ 1 callersMethodclose_writer
(self)
dassl/engine/trainer.py:230
↓ 1 callersFunctioncollect_env_info
Return env info as a string. Code source: github.com/facebookresearch/maskrcnn-benchmark
dassl/utils/tools.py:123
↓ 1 callersFunctioncompute_ci95
(res)
tools/parse_test_res.py:60
↓ 1 callersFunctioncosine_distance
Computes cosine distance. Args: input1 (torch.Tensor): 2-D feature matrix. input2 (torch.Tensor): 2-D feature matrix. Return
dassl/metrics/distance.py:64
↓ 1 callersMethoddecode
Decodes a list of string notations to specify blocks inside the network. :param string_list: a list of strings, each string is a not
dassl/modeling/backbone/efficientnet/utils.py:334
↓ 1 callersMethoddetect_anomaly
(self, loss)
dassl/engine/trainer.py:221
↓ 1 callersMethoddistance
(batch1, batch2, dist_metric="cosine")
dassl/modeling/ops/optimal_transport.py:9
↓ 1 callersMethoddomain_mix
(self, x, target, domain)
dassl/engine/dg/domain_mix.py:56
↓ 1 callersFunctiondownload_and_prepare
(root)
datasets/ssl/stl10.py:27
↓ 1 callersFunctiondrop_connect
Drop connect.
dassl/modeling/backbone/efficientnet/utils.py:106
↓ 1 callersFunctionefficientnet
Creates a efficientnet model.
dassl/modeling/backbone/efficientnet/utils.py:361
↓ 1 callersFunctioneuclidean_squared_distance
Computes euclidean squared distance. Args: input1 (torch.Tensor): 2-D feature matrix. input2 (torch.Tensor): 2-D feature matrix.
dassl/metrics/distance.py:46
↓ 1 callersMethodevaluate
(self)
dassl/evaluation/evaluator.py:22
↓ 1 callersFunctionextend_cfg
Add new config variables. E.g. from yacs.config import CfgNode as CN cfg.TRAINER.MY_MODEL = CN() cfg.TRAINER.MY_MODE
tools/train.py:55
↓ 1 callersFunctionextract_and_save
(images, labels, level, dst)
datasets/dg/cifar_c.py:15
↓ 1 callersMethodextract_features
Returns output of the final convolution layer
dassl/modeling/backbone/efficientnet/model.py:246
↓ 1 callersMethodfeaturemaps
(self, x)
dassl/modeling/backbone/resnet.py:190
↓ 1 callersFunctionfind_version
()
setup.py:12
↓ 1 callersFunctionfixmatch_list
()
dassl/data/transforms/randaugment.py:289
↓ 1 callersMethodflush
(self)
dassl/utils/logger.py:48
↓ 1 callersMethodfrom_pretrained
( cls, model_name, advprop=False, num_classes=1000, in_channels=3 )
dassl/modeling/backbone/efficientnet/model.py:289
↓ 1 callersMethodgenerate_fewshot_dataset
Generate a few-shot dataset (typically for the training set). This function is useful when one wants to evaluate a model in a few-sho
dassl/data/datasets/base_dataset.py:167
↓ 1 callersFunctionget_cfg_default
()
dassl/config/__init__.py:4
↓ 1 callersMethodget_domain
(self, dataset, idx)
dassl/data/datasets/dg/wilds/wilds_base.py:72
↓ 1 callersMethodget_image_path
(self, dataset, idx)
dassl/data/datasets/dg/wilds/wilds_base.py:64
↓ 1 callersMethodget_lab2cname
Get a label-to-classname mapping (dict). Args: data_source (list): a list of Datum objects.
dassl/data/datasets/base_dataset.py:107
↓ 1 callersMethodget_label
(self, dataset, idx)
dassl/data/datasets/dg/wilds/wilds_base.py:69
↓ 1 callersFunctionget_model_params
Get the block args and global params for a given model
dassl/modeling/backbone/efficientnet/utils.py:399
↓ 1 callersFunctionget_most_similar_str_to_a_from_b
Return the most similar string to a in b. Args: a (str): probe string. b (list): a list of candidate strings.
dassl/utils/tools.py:148
↓ 1 callersMethodget_num_classes
Count number of classes. Args: data_source (list): a list of Datum objects.
dassl/data/datasets/base_dataset.py:95
↓ 1 callersFunctionget_requirements
(filename='requirements.txt')
setup.py:27
↓ 1 callersMethodget_similarity_matrix
(feat, topk, device)
dassl/engine/da/cdac.py:253
↓ 1 callersFunctionget_width_and_height_from_size
Obtains width and height from a int or tuple
dassl/modeling/backbone/efficientnet/utils.py:130
↓ 1 callersFunctioninit_pretrained_weights
(model, model_url)
dassl/modeling/backbone/alexnet.py:52
← previousnext →101–200 of 664, ranked by callers