MCPcopy Create free account

hub / github.com/LeapLabTHU/DAT / functions

Functions70 in github.com/LeapLabTHU/DAT

⨍ Functions 70◇ Types & classes 15
↓ 6 callersMethod__init__
(self, dim, heads, window_size, attn_drop, proj_drop)
models/dat_blocks.py:21
↓ 6 callersFunctionreduce_tensor
(tensor)
utils.py:91
↓ 4 callersFunctionget_grad_norm
(parameters, norm_type=2)
utils.py:65
↓ 3 callersFunctioncheck_keywords_in_name
(name, keywords=())
optimizer.py:95
↓ 2 callersMethod__init__
(self, img_size=224, patch_size=4, num_classes=1000, expansion=4, dim_stem=96, dims=[96, 192,
models/dat.py:164
↓ 2 callersMethod_get_lr
(self, t)
lr_scheduler.py:85
↓ 2 callersMethod_get_q_grid
(self, H, W, B, dtype, device)
models/dat_blocks.py:234
↓ 2 callersFunctionauto_resume_helper
(output_dir)
utils.py:78
↓ 2 callersFunctionbuild_dataset
(is_train, config)
data/build.py:66
↓ 2 callersFunctionbuild_loader
(config)
data/build.py:21
↓ 2 callersFunctionbuild_model
(config)
models/build.py:13
↓ 2 callersFunctionbuild_optimizer
Build optimizer, set weight decay of normalization to 0 by default.
optimizer.py:13
↓ 2 callersFunctionbuild_scheduler
(config, optimizer, n_iter_per_epoch)
lr_scheduler.py:17
↓ 2 callersFunctioncreate_logger
(output_dir, dist_rank=0, name='')
logger.py:18
↓ 2 callersFunctionget_config
Get a yacs CfgNode object with default values.
config.py:204
↓ 2 callersFunctionload_checkpoint
(config, model, optimizer, lr_scheduler, logger)
utils.py:17
↓ 2 callersFunctionload_pretrained
(ckpt_path, model, logger)
utils.py:41
↓ 2 callersFunctionsave_checkpoint
(config, epoch, model, max_accuracy, optimizer, lr_scheduler, logger)
utils.py:51
↓ 2 callersMethodset_epoch
(self, epoch)
data/samplers.py:30
↓ 2 callersFunctionvalidate
(config, data_loader, model, logger)
slurm_main.py:253
↓ 2 callersFunctionvalidate
(config, data_loader, model, logger)
main.py:259
↓ 1 callersMethod_get_ref_points
(self, H_key, W_key, B, dtype, device)
models/dat_blocks.py:219
↓ 1 callersFunction_update_config_from_file
(config, cfg_file)
config.py:156
↓ 1 callersFunctionbuild_transform
(is_train, config)
data/build.py:79
↓ 1 callersMethodforward
(self, x, mask=None)
models/dat_blocks.py:55
↓ 1 callersFunctioninit_dist_slurm
Initialize slurm distributed training environment. If argument ``port`` is not specified, then the master port will be system environment vari
utils.py:97
↓ 1 callersMethodload_pretrained
(self, state_dict, lookup_22k)
models/dat.py:254
↓ 1 callersFunctionmain
()
slurm_main.py:67
↓ 1 callersFunctionmain
()
main.py:67
↓ 1 callersMethodno_weight_decay
(self)
models/dat.py:297
↓ 1 callersMethodno_weight_decay_keywords
(self)
models/dat.py:301
↓ 1 callersFunctionparse_option
()
slurm_main.py:39
↓ 1 callersFunctionparse_option
()
main.py:39
↓ 1 callersMethodreset_parameters
(self)
models/slide.py:56
↓ 1 callersMethodreset_parameters
(self)
models/dat.py:246
↓ 1 callersFunctionset_weight_decay_and_lr
( model, skip_list=(), skip_keywords=(), lower_lr_kvs={}, base_lr=5e-4)
optimizer.py:44
↓ 1 callersFunctionthroughput
(data_loader, model, logger)
slurm_main.py:299
↓ 1 callersFunctionthroughput
(data_loader, model, logger)
main.py:305
↓ 1 callersFunctiontrain_one_epoch
(config, model, criterion, data_loader, optimizer, epoch, mixup_fn, lr_scheduler, logger)
slurm_main.py:182
↓ 1 callersFunctiontrain_one_epoch
(config, model, criterion, data_loader, optimizer, epoch, mixup_fn, lr_scheduler, logger)
main.py:188
↓ 1 callersFunctionupdate_config
(config, args)
config.py:171
Method__init__
(self, optimizer: torch.optim.Optimizer, t_initial: int, lr
lr_scheduler.py:56
Method__init__
(self, indices)
data/samplers.py:20
Method__init__
( self, dim, num_heads, ka, qkv_bias=True, qk_scale=None, attn_drop=0., proj_drop=0.,dim_reduc
models/slide.py:17
Method__init__
(self, dim, kernel_size, num_heads, attn_drop=0., proj_drop=0., dilation=None)
models/nat.py:40
Method__init__
(self, dim, heads, window_size, attn_drop, proj_drop, shift_size, fmap_size)
models/dat_blocks.py:95
Method__init__
( self, q_size, kv_size, n_heads, n_head_channels, n_groups, attn_drop, proj_drop, stride,
models/dat_blocks.py:132
Method__init__
(self, dim, num_heads=8, attn_drop=0., proj_drop=0., sr_ratio=1)
models/dat_blocks.py:336
Method__init__
(self, channels, expansion, drop)
models/dat_blocks.py:392
Method__init__
(self, dim)
models/dat_blocks.py:415
Method__init__
(self, channels, expansion, drop)
models/dat_blocks.py:429
Method__init__
(self, n_q, n_channels, n_heads, ksize, stride, padding, qna_activation)
models/qna.py:19
Method__init__
(self, dim: int, inplace: bool = False, init_values: float
models/dat.py:26
Method__init__
(self, fmap_size, window_size, ns_per_pt, dim_in, dim_embed, depths, stage_spec, n_groups,
models/dat.py:42
Method__iter__
(self)
data/samplers.py:24
Method__len__
(self)
data/samplers.py:27
Methodforward
(self, x)
models/slide.py:64
Methodforward
(self, x)
models/nat.py:68
Methodforward
(self, x)
models/dat_blocks.py:121
Methodforward
(self, x)
models/dat_blocks.py:248
Methodforward
(self, x)
models/dat_blocks.py:361
Methodforward
(self, x)
models/dat_blocks.py:405
Methodforward
(self, x)
models/dat_blocks.py:420
Methodforward
(self, x)
models/dat_blocks.py:450
Methodforward
(self, x)
models/qna.py:45
Methodforward
(self, x)
models/dat.py:34
Methodforward
(self, x)
models/dat.py:133
Methodforward
(self, x)
models/dat.py:304
Methodget_epoch_values
(self, epoch: int)
lr_scheduler.py:94
Methodget_update_values
(self, num_updates: int)
lr_scheduler.py:100