MCPcopy Create free account

hub / github.com/LayneH/GreenMIM / functions

Functions148 in github.com/LayneH/GreenMIM

⨍ Functions 148◇ Types & classes 34
Functionall_reduce_mean
(x)
util/misc.py:349
Methodavg
(self)
util/misc.py:61
Functionbuild_dataset
(is_train, args)
util/datasets.py:22
Methodextra_repr
(self)
modeling/green_swin_models.py:96
Methodextra_repr
(self)
modeling/green_swin_models.py:150
Methodextra_repr
(self)
modeling/green_swin_models.py:241
Methodextra_repr
(self)
modeling/group_window_attention.py:107
Methodforward
(self, x: torch.Tensor, indexes: torch.Tensor, H: int, W: int)
modeling/sparse_conv_spconv.py:24
Methodforward
(self, x: torch.Tensor, indexes: torch.Tensor, mask: torch.Tensor, H: int, W: int)
modeling/sparse_conv_spconv.py:53
Methodforward
(self, x)
modeling/green_swin_models.py:30
Methodforward
(self, x, attn_mask, rel_pos_idx)
modeling/green_swin_models.py:83
Methodforward
x: B, H*W, C
modeling/green_swin_models.py:117
Methodforward
(self, x, coords, patch_mask)
modeling/green_swin_models.py:210
Methodforward
(self, x)
modeling/green_swin_models.py:276
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
modeling/group_window_attention.py:67
Methodforward
(self, x, mask)
modeling/base_green_models.py:52
Methodforward
(self, imgs, mask_ratio=0.75)
modeling/base_green_models.py:232
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
modeling/green_twins_models.py:19
Methodforward
(self, x, H, W, vis_coords)
modeling/green_twins_models.py:72
Methodforward
(self, x, H, W)
modeling/green_twins_models.py:123
Methodforward
(self, x, H, W, vis_coords=None, attn_mask=None)
modeling/green_twins_models.py:138
Methodforward
(self, x)
modeling/green_twins_models.py:165
Methodforward
(self, x: torch.Tensor, indexes: torch.Tensor, mask: torch.Tensor, H: int, W: int)
modeling/green_twins_models.py:193
Methodforward
(self, x, indexes, vis_mask, H, W)
modeling/green_twins_models.py:285
Methodforward
(self, x: torch.Tensor, indexes: torch.Tensor, H: int = None, W: int = None)
modeling/sparse_conv_me.py:53
Methodforward_features
(self, x, mask)
modeling/green_swin_models.py:395
Methodforward_features
(self, x)
modeling/green_twins_models.py:274
Methodforward_features
(self, x_, mask)
modeling/green_twins_models.py:340
Methodglobal_avg
(self)
util/misc.py:66
Functiongreen_mim_swin_base_patch4_dec512b1
(**kwargs)
modeling/model_factory.py:25
Functiongreen_mim_swin_base_patch4_win14_dec512b1
(**kwargs)
modeling/model_factory.py:64
Functiongreen_mim_swin_large_patch4_win14_dec512b1
(**kwargs)
modeling/model_factory.py:103
Functiongreen_mim_twins_large_patch4_dec512b1
(**kwargs)
modeling/model_factory.py:142
Functioninit_distributed_mode
(args)
util/misc.py:216
Functioninterpolate_pos_embed
(model, checkpoint_model)
util/pos_embed.py:75
Functionload_model
(args, model_without_ddp, optimizer, loss_scaler)
util/misc.py:332
Methodmedian
(self)
util/misc.py:56
Methodno_weight_decay
(self)
modeling/green_swin_models.py:380
Methodno_weight_decay
(self)
modeling/green_twins_models.py:289
Methodno_weight_decay
(self)
modeling/green_twins_models.py:332
Methodno_weight_decay_keywords
(self)
modeling/green_swin_models.py:384
Functionparam_groups_lrd
Parameter groups for layer-wise lr decay Following BEiT: https://github.com/microsoft/unilm/blob/master/beit/optim_factory.py#L58
util/lr_decay.py:15
Methodpatchify
(self, x)
modeling/base_green_models.py:46
Methodreset_drop_path
(self, drop_path_rate)
modeling/green_twins_models.py:257
Functionsave_model
(args, epoch, model, model_without_ddp, optimizer, loss_scaler)
util/misc.py:312
Methodsynchronize_between_processes
Warning: does not synchronize the deque!
util/misc.py:42
Methodunpatchify
x: (N, L, patch_size**2 *3) imgs: (N, 3, H, W)
modeling/base_green_models.py:140
Methodvalue
(self)
util/misc.py:74
← previous101–148 of 148, ranked by callers