Functions — github.com/SwinTransformer/Transformer-SSL

hub / github.com/SwinTransformer/Transformer-SSL / functions

Functions101 in github.com/SwinTransformer/Transformer-SSL

⨍ Functions 101◇ Types & classes 16

↓ 10 callersFunctionget_grad_norm

(parameters, norm_type=2)

utils.py:81

↓ 6 callersMethod__init__

(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)

models/swin_transformer.py:16

↓ 6 callersMethodflops

(self)

models/swin_transformer.py:409

↓ 6 callersFunctionreduce_tensor

(tensor)

utils.py:110

↓ 4 callersMethodget_zipfile

(path)

data/zipreader.py:31

↓ 4 callersMethodread

(path)

data/zipreader.py:86

↓ 4 callersMethodsplit_zip_style_path

(path)

data/zipreader.py:39

↓ 3 callersFunctionauto_resume_helper

(output_dir)

utils.py:94

↓ 3 callersFunctionbuild_loader

(config)

data/build.py:25

↓ 3 callersFunctionbuild_model

(config)

models/build.py:20

↓ 3 callersFunctionbuild_optimizer

Build optimizer, set weight decay of normalization to 0 by default.

optimizer.py:11

↓ 3 callersFunctionbuild_scheduler

(config, optimizer, n_iter_per_epoch)

lr_scheduler.py:14

↓ 3 callersFunctioncreate_logger

(output_dir, dist_rank=0, name='')

logger.py:16

↓ 3 callersFunctionget_config

Get a yacs CfgNode object with default values.

config.py:257

↓ 3 callersFunctionload_checkpoint

(config, model, optimizer, lr_scheduler, logger)

utils.py:37

↓ 3 callersFunctionsave_checkpoint

(config, epoch, model, max_accuracy, optimizer, lr_scheduler, logger)

utils.py:64

↓ 3 callersMethodset_epoch

(self, epoch)

data/samplers.py:28

↓ 2 callersMethod_get_lr

(self, t)

lr_scheduler.py:83

↓ 2 callersFunctionbuild_dataset

(is_train, config)

data/build.py:75

↓ 2 callersMethodcontrastive_loss

(self, q, k, queue)

models/moby.py:120

↓ 2 callersFunctiondist_collect

collect all tensor from all GPUs args: x: shape (mini_batch, ...) returns: shape (mini_batch * num_gpu, ...)

models/moby.py:18

↓ 2 callersFunctionpil_loader

(path)

data/cached_image_folder.py:179

↓ 2 callersFunctionvalidate

(config, data_loader, model)

moby_linear.py:270

↓ 2 callersFunctionvalidate

(config, data_loader, model)

main.py:232

↓ 2 callersFunctionwindow_partition

Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)

models/swin_transformer.py:34

↓ 1 callersMethod__init__

(self, root, loader, extensions, ann_file='', img_prefix='', transform=None, target_transform=None,

data/cached_image_folder.py:93

↓ 1 callersMethod__init__

(self, cfg, encoder, encoder_k, contrast_m

models/moby.py:33

↓ 1 callersMethod__len__

(self)

data/cached_image_folder.py:162

↓ 1 callersMethod_dequeue_and_enqueue

(self, keys1, keys2)

models/moby.py:103

↓ 1 callersMethod_momentum_update_key_encoder

Momentum update of the key encoder

models/moby.py:89

↓ 1 callersFunction_update_config_from_file

(config, cfg_file)

config.py:200

↓ 1 callersFunctionaccimage_loader

(path)

data/cached_image_folder.py:192

↓ 1 callersFunctionbuild_transform

(is_train, config)

data/build.py:95

↓ 1 callersFunctioncheck_keywords_in_name

(name, keywords=())

optimizer.py:52

↓ 1 callersFunctionfind_classes

(dir)

data/cached_image_folder.py:30

↓ 1 callersMethodforward_features

(self, x)

models/swin_transformer.py:568

↓ 1 callersFunctionhas_file_allowed_extension

Checks if a file is an allowed extension. Args: filename (string): path to a file Returns: bool: True if the filename ends wit

data/cached_image_folder.py:19

↓ 1 callersMethodinit_cache

(self)

data/cached_image_folder.py:124

↓ 1 callersFunctionis_zip_path

judge if this is a zip path

data/zipreader.py:18

↓ 1 callersFunctionload_pretrained

(model, ckpt_path, logger)

utils.py:20

↓ 1 callersFunctionmain

(config)

moby_main.py:75

↓ 1 callersFunctionmain

(config)

moby_linear.py:106

↓ 1 callersFunctionmain

(config)

main.py:76

↓ 1 callersFunctionmake_dataset

(dir, class_to_idx, extensions)

data/cached_image_folder.py:37

↓ 1 callersFunctionmake_dataset_with_ann

(ann_file, img_prefix, extensions)

data/cached_image_folder.py:55

↓ 1 callersMethodno_weight_decay

(self)

models/swin_transformer.py:561

↓ 1 callersMethodno_weight_decay_keywords

(self)

models/swin_transformer.py:565

↓ 1 callersFunctionparse_option

()

moby_main.py:35

↓ 1 callersFunctionparse_option

()

moby_linear.py:37

↓ 1 callersFunctionparse_option

()

main.py:36

↓ 1 callersFunctionset_weight_decay

(model, skip_list=(), skip_keywords=())

optimizer.py:35

↓ 1 callersFunctionthroughput

(data_loader, model, logger)

moby_linear.py:317

↓ 1 callersFunctionthroughput

(data_loader, model, logger)

main.py:279

↓ 1 callersFunctiontrain_one_epoch

(config, model, data_loader, optimizer, epoch, lr_scheduler)

moby_main.py:130

↓ 1 callersFunctiontrain_one_epoch

(config, model, criterion, data_loader, optimizer, epoch, mixup_fn, lr_scheduler)

moby_linear.py:188

↓ 1 callersFunctiontrain_one_epoch

(config, model, criterion, data_loader, optimizer, epoch, mixup_fn, lr_scheduler)

main.py:150

↓ 1 callersFunctionupdate_config

(config, args)

config.py:215

↓ 1 callersFunctionwindow_reverse

Args: windows: (num_windows*B, window_size, window_size, C) window_size (int): Window size H (int): Height of image

models/swin_transformer.py:49

Args: index (int): Index Returns: tuple: (sample, target) where target is class_index of the target class.

data/cached_image_folder.py:146

Method__getitem__

Args: index (int): Index Returns: tuple: (image, target) where target is class_index of the target class.

data/cached_image_folder.py:238

Method__getitem__

Args: index (int): Index Returns: tuple: (image, target) where target is class_index of the target class.

data/custom_image_folder.py:12

Method__init__

(self, optimizer: torch.optim.Optimizer, t_initial: int, lr

(self, root, ann_file='', img_prefix='', transform=None, target_transform=None, loader=defaul

data/cached_image_folder.py:228

Method__init__

(self, in_dim=256, inner_dim=4096, out_dim=256, num_layers=2)

models/moby.py:171

Method__init__

(self, dim, window_size, num_heads, qkv_bias=True, qk_scale=None, attn_drop=0., proj_drop=0.)

models/swin_transformer.py:80

Method__init__

(self, dim, input_resolution, num_heads, window_size=7, shift_size=0, mlp_ratio=4., qkv_bias=

models/swin_transformer.py:183

Method__init__

(self, input_resolution, dim, norm_layer=nn.LayerNorm)

models/swin_transformer.py:307

Method__init__

(self, dim, input_resolution, depth, num_heads, window_size, mlp_ratio=4., qkv_bias=True, qk_

models/swin_transformer.py:367

Method__init__

(self, img_size=224, patch_size=4, in_chans=3, embed_dim=96, norm_layer=None)

models/swin_transformer.py:429

Method__init__

(self, img_size=224, patch_size=4, in_chans=3, num_classes=1000, embed_dim=96, depths=[2, 2,

models/swin_transformer.py:492

data/cached_image_folder.py:165

Method_init_weights

(self, m)

models/swin_transformer.py:551

Functiondefault_img_loader

(path)

data/cached_image_folder.py:201

Methodextra_repr

(self)

models/swin_transformer.py:147

Methodextra_repr

(self)

models/swin_transformer.py:279

Methodextra_repr

(self)

models/swin_transformer.py:337

Methodextra_repr

(self)

models/swin_transformer.py:406

Methodflops

(self, N)

models/swin_transformer.py:150

Methodflops

(self)

models/swin_transformer.py:283

Methodflops

(self)

models/swin_transformer.py:340

Methodflops

(self)

models/swin_transformer.py:458

Methodflops

(self)

models/swin_transformer.py:587

models/swin_transformer.py:25

Methodforward

Args: x: input features with shape of (num_windows*B, N, C) mask: (0/-inf) mask with shape of (num_windows, Wh*Ww, Wh

models/swin_transformer.py:114

Methodforward

(self, x)

models/swin_transformer.py:240

Methodforward

x: B, H*W, C

models/swin_transformer.py:314

Methodforward

(self, x)

models/swin_transformer.py:396

Methodforward

(self, x)

models/swin_transformer.py:448

Methodforward

(self, x)

models/swin_transformer.py:582

Methodget_epoch_values

(self, epoch: int)

lr_scheduler.py:92

Methodget_update_values

(self, num_updates: int)

(path, extension=None)

data/zipreader.py:67

next →1–100 of 101, ranked by callers