MCPcopy Create free account

hub / github.com/coeusguo/ceit / functions

Functions62 in github.com/coeusguo/ceit

⨍ Functions 62◇ Types & classes 12
↓ 19 callersFunctionprint
(*args, **kwargs)
utils.py:179
↓ 7 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
ceit_model.py:34
↓ 6 callersMethodupdate
(self, **kwargs)
utils.py:85
↓ 3 callersFunctionis_dist_avail_and_initialized
()
utils.py:187
↓ 2 callersFunctionbuild_dataset
(is_train, args)
datasets.py:58
↓ 2 callersFunctionevaluate
(data_loader, model, device)
engine.py:66
↓ 2 callersMethodlog_every
(self, iterable, print_freq, header=None)
utils.py:115
↓ 2 callersMethodsynchronize_between_processes
(self)
utils.py:108
↓ 2 callersMethodupdate
(self, value, n=1)
utils.py:31
↓ 1 callersMethodadd_meter
(self, name, meter)
utils.py:112
↓ 1 callersFunctionbuild_transform
(is_train, args)
datasets.py:80
↓ 1 callersMethodforward_features
(self, x)
ceit_model.py:345
↓ 1 callersFunctionget_args_parser
()
main.py:27
↓ 1 callersFunctionget_rank
()
utils.py:201
↓ 1 callersFunctionis_main_process
()
utils.py:207
↓ 1 callersFunctionmain
(args)
main.py:169
↓ 1 callersMethodset_epoch
(self, epoch)
samplers.py:58
↓ 1 callersFunctionsetup_for_distributed
This function disables printing when not in master process
utils.py:172
↓ 1 callersFunctiontrain_one_epoch
(model: torch.nn.Module, criterion: torch.nn.Module, data_loader: Iterable, optimizer: tor
engine.py:18
Method__getattr__
(self, attr)
utils.py:92
Method__init__
(self, in_chans=3, out_chans=64, kernel_size=7, stride=2)
ceit_model.py:19
Method__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0., kern
ceit_model.py:53
Method__init__
(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0.)
ceit_model.py:102
Method__init__
(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0.)
ceit_model.py:132
Method__init__
(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0., dr
ceit_model.py:166
Method__init__
(self, backbone, img_size=224, patch_size=16, feature_size=None, in_chans=3, embed_dim=768)
ceit_model.py:207
Method__init__
args: - img_size (:obj:`int`): input image size - patch_size (:obj:`int`): patch size - in_chans (:obj:`i
ceit_model.py:245
Method__init__
(self, dataset, num_replicas=None, rank=None, shuffle=True)
samplers.py:16
Method__init__
(self, window_size=20, fmt=None)
utils.py:23
Method__init__
(self, delimiter="\t")
utils.py:81
Method__init__
(self, root, train=True, year=2018, transform=None, target_transform=None, category='name', l
datasets.py:16
Method__iter__
(self)
samplers.py:35
Method__len__
(self)
samplers.py:55
Method__str__
(self)
utils.py:71
Method__str__
(self)
utils.py:100
Method_init_weights
(self, m)
ceit_model.py:325
Function_load_checkpoint_for_ema
Workaround for ModelEma._load_checkpoint to accept an already-loaded object
utils.py:162
Methodavg
(self)
utils.py:55
Functionceit_base_patch16_224
convolutional + pooling stem local enhanced feedforward attention over cls_tokens
ceit_model.py:406
Functionceit_small_patch16_224
convolutional + pooling stem local enhanced feedforward attention over cls_tokens
ceit_model.py:390
Functionceit_small_patch16_384
convolutional + pooling stem local enhanced feedforward attention over cls_tokens
ceit_model.py:438
Functionceit_tiny_patch16_224
convolutional + pooling stem local enhanced feedforward attention over cls_tokens
ceit_model.py:374
Functionceit_tiny_patch16_384
convolutional + pooling stem local enhanced feedforward attention over cls_tokens
ceit_model.py:422
Methodforward
(self, x)
ceit_model.py:26
Methodforward
(self, x)
ceit_model.py:43
Methodforward
(self, x)
ceit_model.py:76
Methodforward
(self, x)
ceit_model.py:115
Methodforward
(self, x)
ceit_model.py:137
Methodforward
(self, x)
ceit_model.py:191
Methodforward
(self, x)
ceit_model.py:236
Methodforward
(self, x)
ceit_model.py:367
Methodget_classifier
(self)
ceit_model.py:338
Functionget_world_size
()
utils.py:195
Methodglobal_avg
(self)
utils.py:60
Functioninit_distributed_mode
(args)
utils.py:216
Methodmax
(self)
utils.py:64
Methodmedian
(self)
utils.py:50
Methodno_weight_decay
(self)
ceit_model.py:335
Methodreset_classifier
(self, num_classes, global_pool='')
ceit_model.py:341
Functionsave_on_master
(*args, **kwargs)
utils.py:211
Methodsynchronize_between_processes
Warning: does not synchronize the deque!
utils.py:36
Methodvalue
(self)
utils.py:68