MCPcopy Create free account

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

Functions664 in github.com/KaiyangZhou/Dassl.pytorch

↓ 1 callersMethodinit_writer
(self, log_dir)
dassl/engine/trainer.py:225
↓ 1 callersMethodlinear_mmd
(self, x, y)
dassl/modeling/ops/mmd.py:28
↓ 1 callersFunctionlinear_rampup
Linear rampup. Args: current (int): current step. rampup_length (int): maximum step.
dassl/modeling/ops/utils.py:56
↓ 1 callersMethodload_classnames
(self)
dassl/data/datasets/dg/wilds/wilds_base.py:61
↓ 1 callersFunctionload_pretrained_weights
Loads pretrained weights, and downloads if loading for the first time.
dassl/modeling/backbone/efficientnet/utils.py:461
↓ 1 callersFunctionload_pretrained_weights
r"""Load pretrianed weights to model. Features:: - Incompatible layers (unmatched in name or size) will be ignored. - Can automat
dassl/utils/torchtools.py:267
↓ 1 callersFunctionload_state_dict
r"""Copies parameters and buffers from :attr:`state_dict` into this module and its descendants. If :attr:`strict` is ``True``, then the keys o
dassl/modeling/backbone/resnet_dynamic.py:81
↓ 1 callersFunctionmain
(args)
tools/train.py:93
↓ 1 callersFunctionmain
()
tools/replace_text.py:38
↓ 1 callersFunctionmain
(args, end_signal)
tools/parse_test_res.py:123
↓ 1 callersFunctionmain
(npy_folder)
datasets/dg/cifar_c.py:29
↓ 1 callersFunctionmain
(data_dir)
datasets/da/digit5.py:98
↓ 1 callersFunctionmake_layers
(cfg, batch_norm=False)
dassl/modeling/backbone/vgg.py:67
↓ 1 callersMethodmodel_backward
(self, loss)
dassl/engine/trainer.py:290
↓ 1 callersMethodmodel_inference
(self, input)
dassl/engine/trainer.py:475
↓ 1 callersMethodmodel_update
(self, names=None)
dassl/engine/trainer.py:294
↓ 1 callersMethodmodel_zero_grad
(self, names=None)
dassl/engine/trainer.py:284
↓ 1 callersMethodmoment_distance
(self, x, u)
dassl/engine/da/m3sda.py:153
↓ 1 callersFunctionopen_specified_layers
r"""Open specified layers in model for training while keeping other layers frozen. Args: model (nn.Module): neural net model.
dassl/utils/torchtools.py:206
↓ 1 callersMethodparse_batch_test
(self, batch)
dassl/engine/trainer.py:478
↓ 1 callersMethodparse_batch_train
(self, batch)
dassl/engine/dg/domain_mix.py:44
↓ 1 callersMethodparse_batch_train
(self, batch)
dassl/engine/dg/vanilla.py:30
↓ 1 callersMethodparse_batch_train
(self, batch_x, batch_u)
dassl/engine/da/source_only.py:29
↓ 1 callersMethodparse_batch_train
(self, batch_x, batch_u)
dassl/engine/da/m3sda.py:189
↓ 1 callersMethodparse_batch_train
(self, batch_x, batch_u)
dassl/engine/da/dael.py:183
↓ 1 callersMethodparse_batch_train
(self, batch_x, batch_u)
dassl/engine/da/cdac.py:229
↓ 1 callersMethodparse_batch_train
(self, batch_x, batch_u)
dassl/engine/ssl/mixmatch.py:88
↓ 1 callersMethodparse_batch_train
(self, batch_x, batch_u)
dassl/engine/ssl/sup_baseline.py:27
↓ 1 callersMethodpoly_mmd
(self, x, y, alpha=1.0, c=2.0, d=2)
dassl/modeling/ops/mmd.py:35
↓ 1 callersFunctionprint_args
(args, cfg)
tools/train.py:9
↓ 1 callersMethodprocess
(self, mo, gt)
dassl/evaluation/evaluator.py:19
↓ 1 callersFunctionrandaugment_list
()
dassl/data/transforms/randaugment.py:221
↓ 1 callersFunctionrandaugment_list2
()
dassl/data/transforms/randaugment.py:267
↓ 1 callersMethodrbf_mmd
(self, x, y)
dassl/modeling/ops/mmd.py:45
↓ 1 callersFunctionread_image
Read image from path using ``PIL.Image``. Args: path (str): path to an image. Returns: PIL image
dassl/utils/tools.py:111
↓ 1 callersFunctionreadme
()
setup.py:6
↓ 1 callersFunctionrecursive_update
(directory, text_to_search, replacement_text)
tools/replace_text.py:24
↓ 1 callersMethodreset
(self)
dassl/utils/meters.py:25
↓ 1 callersMethodreset
(self)
dassl/evaluation/evaluator.py:16
↓ 1 callersFunctionreset_cfg
(cfg, args)
tools/train.py:23
↓ 1 callersFunctionresume_from_checkpoint
r"""Resume training from a checkpoint. This will load (1) model weights and (2) ``state_dict`` of optimizer if ``optimizer`` is not None.
dassl/utils/torchtools.py:118
↓ 1 callersMethodresume_model_if_exist
(self, directory)
dassl/engine/trainer.py:147
↓ 1 callersFunctionround_repeats
Round number of filters based on depth multiplier.
dassl/modeling/backbone/efficientnet/utils.py:98
↓ 1 callersMethodrun_epoch
(self)
dassl/engine/trainer.py:266
↓ 1 callersFunctionsave_checkpoint
r"""Save checkpoint. Args: state (dict): dictionary. save_dir (str): directory to save checkpoint. is_best (bool, optiona
dassl/utils/torchtools.py:27
↓ 1 callersFunctionset_random_seed
(seed)
dassl/utils/tools.py:73
↓ 1 callersMethodset_swish
Sets swish function as memory efficient (for training) or standard (for export)
dassl/modeling/backbone/efficientnet/model.py:137
↓ 1 callersFunctionsetup_cfg
(args)
tools/train.py:69
↓ 1 callersFunctionsetup_logger
(output=None)
dassl/utils/logger.py:60
↓ 1 callersFunctionsharpen_prob
Sharpening probability with a temperature. Args: p (torch.Tensor): probability matrix (batch_size, n_classes) temperature (float)
dassl/modeling/ops/utils.py:5
↓ 1 callersMethodshow_dataset_summary
(self, cfg)
dassl/data/data_manager.py:166
↓ 1 callersFunctionshuffle_index
Shuffle order.
dassl/modeling/ops/utils.py:22
↓ 1 callersMethodsigmoid_rampup
Exponential Rampup https://arxiv.org/abs/1610.02242
dassl/engine/da/cdac.py:266
↓ 1 callersMethodsinkhorn_iterate
(C, eps, max_iter, thre)
dassl/modeling/ops/optimal_transport.py:69
↓ 1 callersMethodsplit_dataset_by_domain
Split a dataset, i.e. a list of Datum objects, into domain-specific groups stored in a dictionary. Args: data_source (lis
dassl/data/datasets/base_dataset.py:225
↓ 1 callersMethodsplit_dataset_by_label
Split a dataset, i.e. a list of Datum objects, into class-specific groups stored in a dictionary. Args: data_source (list
dassl/data/datasets/base_dataset.py:211
↓ 1 callersMethodstn
Spatial transformer network.
dassl/modeling/network/ddaig_fcn.py:244
↓ 1 callersMethodtest
(self)
dassl/engine/trainer.py:269
↓ 1 callersFunctiontolist_if_not
Convert to a list.
dassl/utils/tools.py:181
FunctionAutoContrast
(img, _)
dassl/data/transforms/randaugment.py:71
FunctionBrightness
(img, v)
dassl/data/transforms/randaugment.py:117
FunctionColor
(img, v)
dassl/data/transforms/randaugment.py:112
FunctionContrast
(img, v)
dassl/data/transforms/randaugment.py:107
FunctionCutout
(img, v)
dassl/data/transforms/randaugment.py:127
FunctionEqualize
(img, _)
dassl/data/transforms/randaugment.py:79
FunctionFlip
(img, _)
dassl/data/transforms/randaugment.py:83
FunctionIdentity
(img, v)
dassl/data/transforms/randaugment.py:169
FunctionInvert
(img, _)
dassl/data/transforms/randaugment.py:75
MethodM
Modified cost for logarithmic updates. Eq: M_{ij} = (-c_{ij} + u_i + v_j) / epsilon
dassl/modeling/ops/optimal_transport.py:76
FunctionPosterize
(img, v)
dassl/data/transforms/randaugment.py:101
FunctionRotate
(img, v)
dassl/data/transforms/randaugment.py:64
FunctionSamplePairing
(imgs)
dassl/data/transforms/randaugment.py:159
FunctionSharpness
(img, v)
dassl/data/transforms/randaugment.py:122
FunctionShearX
(img, v)
dassl/data/transforms/randaugment.py:16
FunctionShearY
(img, v)
dassl/data/transforms/randaugment.py:23
FunctionSolarize
(img, v)
dassl/data/transforms/randaugment.py:87
FunctionSolarizeAdd
(img, addition=0, threshold=128)
dassl/data/transforms/randaugment.py:92
FunctionTranslateX
(img, v)
dassl/data/transforms/randaugment.py:30
FunctionTranslateXabs
(img, v)
dassl/data/transforms/randaugment.py:39
FunctionTranslateY
(img, v)
dassl/data/transforms/randaugment.py:47
FunctionTranslateYabs
(img, v)
dassl/data/transforms/randaugment.py:56
Method__call__
(self, img)
dassl/data/transforms/transforms.py:64
Method__call__
(self, img)
dassl/data/transforms/transforms.py:107
Method__call__
Args: img (Tensor): tensor image of size (C, H, W). Returns: Tensor: image with n_holes of dimension
dassl/data/transforms/transforms.py:131
Method__call__
(self, img)
dassl/data/transforms/transforms.py:169
Method__call__
(self, img)
dassl/data/transforms/randaugment.py:181
Method__call__
(self, img)
dassl/data/transforms/randaugment.py:203
Method__call__
(self, img)
dassl/data/transforms/randaugment.py:319
Method__call__
(self, img)
dassl/data/transforms/randaugment.py:336
Method__call__
(self, img)
dassl/data/transforms/randaugment.py:355
Method__call__
(self, img)
dassl/data/transforms/autoaugment.py:52
Method__call__
(self, img)
dassl/data/transforms/autoaugment.py:103
Method__call__
(self, img)
dassl/data/transforms/autoaugment.py:154
Method__call__
(self, img)
dassl/data/transforms/autoaugment.py:268
Method__del__
(self)
dassl/utils/logger.py:34
Method__enter__
(self)
dassl/utils/logger.py:37
Method__exit__
(self, *args)
dassl/utils/logger.py:40
Method__getitem__
(self, idx)
dassl/data/data_manager.py:220
Method__init__
( self, dist_metric="cosine", eps=0.01, max_iter=5, bp_to_sinkhorn=Fal
dassl/modeling/ops/optimal_transport.py:105
Method__init__
(self, num_features, n_domain)
dassl/modeling/ops/dsbn.py:38
← previousnext →201–300 of 664, ranked by callers