Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/KaiyangZhou/Dassl.pytorch
/ functions
Functions
664 in github.com/KaiyangZhou/Dassl.pytorch
⨍
Functions
664
◇
Types & classes
136
↓ 1 callers
Method
init_writer
(self, log_dir)
dassl/engine/trainer.py:225
↓ 1 callers
Method
linear_mmd
(self, x, y)
dassl/modeling/ops/mmd.py:28
↓ 1 callers
Function
linear_rampup
Linear rampup. Args: current (int): current step. rampup_length (int): maximum step.
dassl/modeling/ops/utils.py:56
↓ 1 callers
Method
load_classnames
(self)
dassl/data/datasets/dg/wilds/wilds_base.py:61
↓ 1 callers
Function
load_pretrained_weights
Loads pretrained weights, and downloads if loading for the first time.
dassl/modeling/backbone/efficientnet/utils.py:461
↓ 1 callers
Function
load_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 callers
Function
load_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 callers
Function
main
(args)
tools/train.py:93
↓ 1 callers
Function
main
()
tools/replace_text.py:38
↓ 1 callers
Function
main
(args, end_signal)
tools/parse_test_res.py:123
↓ 1 callers
Function
main
(npy_folder)
datasets/dg/cifar_c.py:29
↓ 1 callers
Function
main
(data_dir)
datasets/da/digit5.py:98
↓ 1 callers
Function
make_layers
(cfg, batch_norm=False)
dassl/modeling/backbone/vgg.py:67
↓ 1 callers
Method
model_backward
(self, loss)
dassl/engine/trainer.py:290
↓ 1 callers
Method
model_inference
(self, input)
dassl/engine/trainer.py:475
↓ 1 callers
Method
model_update
(self, names=None)
dassl/engine/trainer.py:294
↓ 1 callers
Method
model_zero_grad
(self, names=None)
dassl/engine/trainer.py:284
↓ 1 callers
Method
moment_distance
(self, x, u)
dassl/engine/da/m3sda.py:153
↓ 1 callers
Function
open_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 callers
Method
parse_batch_test
(self, batch)
dassl/engine/trainer.py:478
↓ 1 callers
Method
parse_batch_train
(self, batch)
dassl/engine/dg/domain_mix.py:44
↓ 1 callers
Method
parse_batch_train
(self, batch)
dassl/engine/dg/vanilla.py:30
↓ 1 callers
Method
parse_batch_train
(self, batch_x, batch_u)
dassl/engine/da/source_only.py:29
↓ 1 callers
Method
parse_batch_train
(self, batch_x, batch_u)
dassl/engine/da/m3sda.py:189
↓ 1 callers
Method
parse_batch_train
(self, batch_x, batch_u)
dassl/engine/da/dael.py:183
↓ 1 callers
Method
parse_batch_train
(self, batch_x, batch_u)
dassl/engine/da/cdac.py:229
↓ 1 callers
Method
parse_batch_train
(self, batch_x, batch_u)
dassl/engine/ssl/mixmatch.py:88
↓ 1 callers
Method
parse_batch_train
(self, batch_x, batch_u)
dassl/engine/ssl/sup_baseline.py:27
↓ 1 callers
Method
poly_mmd
(self, x, y, alpha=1.0, c=2.0, d=2)
dassl/modeling/ops/mmd.py:35
↓ 1 callers
Function
print_args
(args, cfg)
tools/train.py:9
↓ 1 callers
Method
process
(self, mo, gt)
dassl/evaluation/evaluator.py:19
↓ 1 callers
Function
randaugment_list
()
dassl/data/transforms/randaugment.py:221
↓ 1 callers
Function
randaugment_list2
()
dassl/data/transforms/randaugment.py:267
↓ 1 callers
Method
rbf_mmd
(self, x, y)
dassl/modeling/ops/mmd.py:45
↓ 1 callers
Function
read_image
Read image from path using ``PIL.Image``. Args: path (str): path to an image. Returns: PIL image
dassl/utils/tools.py:111
↓ 1 callers
Function
readme
()
setup.py:6
↓ 1 callers
Function
recursive_update
(directory, text_to_search, replacement_text)
tools/replace_text.py:24
↓ 1 callers
Method
reset
(self)
dassl/utils/meters.py:25
↓ 1 callers
Method
reset
(self)
dassl/evaluation/evaluator.py:16
↓ 1 callers
Function
reset_cfg
(cfg, args)
tools/train.py:23
↓ 1 callers
Function
resume_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 callers
Method
resume_model_if_exist
(self, directory)
dassl/engine/trainer.py:147
↓ 1 callers
Function
round_repeats
Round number of filters based on depth multiplier.
dassl/modeling/backbone/efficientnet/utils.py:98
↓ 1 callers
Method
run_epoch
(self)
dassl/engine/trainer.py:266
↓ 1 callers
Function
save_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 callers
Function
set_random_seed
(seed)
dassl/utils/tools.py:73
↓ 1 callers
Method
set_swish
Sets swish function as memory efficient (for training) or standard (for export)
dassl/modeling/backbone/efficientnet/model.py:137
↓ 1 callers
Function
setup_cfg
(args)
tools/train.py:69
↓ 1 callers
Function
setup_logger
(output=None)
dassl/utils/logger.py:60
↓ 1 callers
Function
sharpen_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 callers
Method
show_dataset_summary
(self, cfg)
dassl/data/data_manager.py:166
↓ 1 callers
Function
shuffle_index
Shuffle order.
dassl/modeling/ops/utils.py:22
↓ 1 callers
Method
sigmoid_rampup
Exponential Rampup https://arxiv.org/abs/1610.02242
dassl/engine/da/cdac.py:266
↓ 1 callers
Method
sinkhorn_iterate
(C, eps, max_iter, thre)
dassl/modeling/ops/optimal_transport.py:69
↓ 1 callers
Method
split_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 callers
Method
split_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 callers
Method
stn
Spatial transformer network.
dassl/modeling/network/ddaig_fcn.py:244
↓ 1 callers
Method
test
(self)
dassl/engine/trainer.py:269
↓ 1 callers
Function
tolist_if_not
Convert to a list.
dassl/utils/tools.py:181
Function
AutoContrast
(img, _)
dassl/data/transforms/randaugment.py:71
Function
Brightness
(img, v)
dassl/data/transforms/randaugment.py:117
Function
Color
(img, v)
dassl/data/transforms/randaugment.py:112
Function
Contrast
(img, v)
dassl/data/transforms/randaugment.py:107
Function
Cutout
(img, v)
dassl/data/transforms/randaugment.py:127
Function
Equalize
(img, _)
dassl/data/transforms/randaugment.py:79
Function
Flip
(img, _)
dassl/data/transforms/randaugment.py:83
Function
Identity
(img, v)
dassl/data/transforms/randaugment.py:169
Function
Invert
(img, _)
dassl/data/transforms/randaugment.py:75
Method
M
Modified cost for logarithmic updates. Eq: M_{ij} = (-c_{ij} + u_i + v_j) / epsilon
dassl/modeling/ops/optimal_transport.py:76
Function
Posterize
(img, v)
dassl/data/transforms/randaugment.py:101
Function
Rotate
(img, v)
dassl/data/transforms/randaugment.py:64
Function
SamplePairing
(imgs)
dassl/data/transforms/randaugment.py:159
Function
Sharpness
(img, v)
dassl/data/transforms/randaugment.py:122
Function
ShearX
(img, v)
dassl/data/transforms/randaugment.py:16
Function
ShearY
(img, v)
dassl/data/transforms/randaugment.py:23
Function
Solarize
(img, v)
dassl/data/transforms/randaugment.py:87
Function
SolarizeAdd
(img, addition=0, threshold=128)
dassl/data/transforms/randaugment.py:92
Function
TranslateX
(img, v)
dassl/data/transforms/randaugment.py:30
Function
TranslateXabs
(img, v)
dassl/data/transforms/randaugment.py:39
Function
TranslateY
(img, v)
dassl/data/transforms/randaugment.py:47
Function
TranslateYabs
(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
← previous
next →
201–300 of 664, ranked by callers