MCPcopy Create free account

hub / github.com/Hao840/OFAKD / functions

Functions201 in github.com/Hao840/OFAKD

⨍ Functions 201◇ Types & classes 48
Method_init_weights
(self, m)
custom_model/vip.py:183
Functionbeitv2_base_patch16_224
(pretrained=False, **kwargs)
custom_model/beitv2.py:8
Functionbeitv2_large_patch16_224
(pretrained=False, **kwargs)
custom_model/beitv2.py:17
Methodcls_init_weights
(self, m)
custom_model/cycle_mlp.py:298
Methodcls_init_weights
(self, m)
custom_model/hire_mlp.py:265
Methodconv_bn
(inp, oup, stride)
custom_model/mobilenetv1.py:13
Methodconv_dw
(inp, oup, stride)
custom_model/mobilenetv1.py:20
Methodcuda
(self, *args, **kwargs)
distillers/crd.py:126
Methodextra_repr
(self)
custom_model/cycle_mlp.py:133
Functionforward
(self, x, requires_feat=False)
custom_forward/vip.py:8
Functionforward
(self, x, requires_feat=False)
custom_forward/cycle_mlp.py:8
Functionforward
(self, x, requires_feat=False)
custom_forward/hire_mlp.py:8
Functionforward
(self, x, requires_feat=False)
custom_forward/beit.py:27
Functionforward
(self, x, requires_feat=False)
custom_forward/convnext.py:8
Functionforward
(self, x, requires_feat=False)
custom_forward/swin_transformer.py:8
Functionforward
(self, x, requires_feat=False)
custom_forward/mobilenetv1.py:10
Functionforward
(self, x, requires_feat=False)
custom_forward/mlp_mixer.py:8
Functionforward
(self, x, requires_feat=False)
custom_forward/efficientnet.py:8
Functionforward
(self, x, requires_feat=False)
custom_forward/resnet.py:29
Functionforward
(self, x, requires_feat=False)
custom_forward/vision_transformer.py:10
Methodforward
(self, image, label, *args, **kwargs)
distillers/fitnet.py:33
Methodforward
(self, image, label, *args, **kwargs)
distillers/crd.py:27
Methodforward
(self, x)
distillers/crd.py:58
Methodforward
(self, x)
distillers/crd.py:72
Methodforward
(self, x)
distillers/crd.py:86
Methodforward
(self, v1, v2, y, idx=None)
distillers/crd.py:130
Methodforward
(self, image, label, *args, **kwargs)
distillers/_base.py:14
Methodforward
(self, image, label, *args, **kwargs)
distillers/_base.py:39
Methodforward
(self, image, label, *args, **kwargs)
distillers/ofa.py:108
Methodforward
(self, image, label, *args, **kwargs)
distillers/rkd.py:15
Methodforward
x: B, H*W, C
distillers/utils.py:20
Methodforward
(self, x)
distillers/utils.py:49
Methodforward
(self, x)
distillers/utils.py:62
Methodforward
(self, x)
distillers/utils.py:84
Methodforward
(self, x)
distillers/utils.py:104
Methodforward
(self, x)
distillers/utils.py:113
Methodforward
(self, x)
distillers/utils.py:144
Methodforward
(self, x)
distillers/utils.py:214
Methodforward
(self, image, label, *args, **kwargs)
distillers/correlation.py:19
Methodforward
(self, x)
distillers/correlation.py:44
Methodforward
(self, image, label, *args, **kwargs)
distillers/dist.py:14
Methodforward
(self, image, label, *args, **kwargs)
distillers/kd.py:16
Methodforward
(self, image, label, *args, **kwargs)
distillers/kd.py:39
Methodforward
(self, image, label, *args, **kwargs)
distillers/dkd.py:15
Methodforward
(self, x)
custom_model/vip.py:37
Methodforward
(self, x)
custom_model/vip.py:60
Methodforward
(self, x)
custom_model/vip.py:99
Methodforward
(self, x)
custom_model/vip.py:113
Methodforward
(self, x)
custom_model/vip.py:126
Methodforward
(self, x)
custom_model/vip.py:212
Methodforward
(self, x)
custom_model/cycle_mlp.py:45
Methodforward
Args: input (Tensor[batch_size, in_channels, in_height, in_width]): input tensor
custom_model/cycle_mlp.py:124
Methodforward
(self, x)
custom_model/cycle_mlp.py:160
Methodforward
(self, x)
custom_model/cycle_mlp.py:193
Methodforward
(self, x)
custom_model/cycle_mlp.py:215
Methodforward
(self, x)
custom_model/cycle_mlp.py:229
Methodforward
(self, x)
custom_model/cycle_mlp.py:338
Methodforward
(self, x)
custom_model/hire_mlp.py:35
Methodforward
h: H x W x C -> H/pixel x W x C*pixel w: H x W x C -> H x W/pixel x C*pixel Setting of F.pad: (left, right, top, bottom)
custom_model/hire_mlp.py:74
Methodforward
(self, x)
custom_model/hire_mlp.py:172
Methodforward
(self, x)
custom_model/hire_mlp.py:194
Methodforward
(self, x)
custom_model/hire_mlp.py:212
Methodforward
(self, x)
custom_model/hire_mlp.py:291
Methodforward
(self, x)
custom_model/mobilenetv1.py:49
Functionforward_features
(self, x, requires_feat)
custom_forward/beit.py:10
Functionforward_features
(self, x, requires_feat)
custom_forward/convnext.py:22
Functionforward_features
(self, x, requires_feat)
custom_forward/swin_transformer.py:22
Functionforward_features
(self, x, requires_feat)
custom_forward/mlp_mixer.py:25
Functionforward_features
(self, x, requires_feat)
custom_forward/efficientnet.py:22
Functionforward_features
(self, x, requires_feat)
custom_forward/resnet.py:9
Functionforward_features
(self, x, requires_feat)
custom_forward/vision_transformer.py:24
Methodget_classifier
(self)
custom_model/vip.py:192
Methodget_classifier
(self)
custom_model/cycle_mlp.py:310
Functionhire_mlp_base
(pretrained=False, pretrained_cfg=None, **kwargs)
custom_model/hire_mlp.py:338
Functionhire_mlp_large
(pretrained=False, pretrained_cfg=None, **kwargs)
custom_model/hire_mlp.py:357
Functionhire_mlp_small
(pretrained=False, pretrained_cfg=None, **kwargs)
custom_model/hire_mlp.py:319
Functionhire_mlp_tiny
(pretrained=False, pretrained_cfg=None, **kwargs)
custom_model/hire_mlp.py:300
Functioninit_weights
(module)
distillers/utils.py:148
Functionmobilenetv1
(pretrained=False, **kwargs)
custom_model/mobilenetv1.py:61
Functionpatchify
imgs: (N, C, H, W) x: (N, L, patch_size**2, C)
distillers/utils.py:194
Functionregister_distiller
(fn)
distillers/registry.py:7
Functionregister_method
(fn)
custom_forward/registry.py:8
Methodreset_classifier
(self, num_classes, global_pool='')
custom_model/vip.py:195
Methodreset_classifier
(self, num_classes, global_pool='')
custom_model/cycle_mlp.py:313
Functionstage_info
(self, stage)
custom_forward/vip.py:30
Functionstage_info
(self, stage)
custom_forward/cycle_mlp.py:30
Functionstage_info
(self, stage)
custom_forward/hire_mlp.py:26
Functionstage_info
(self, stage)
custom_forward/beit.py:41
Functionstage_info
(self, stage)
custom_forward/convnext.py:36
Functionstage_info
(self, stage)
custom_forward/swin_transformer.py:40
Functionstage_info
(self, stage)
custom_forward/mobilenetv1.py:28
Functionstage_info
(self, stage)
custom_forward/mlp_mixer.py:39
Functionstage_info
(self, stage)
custom_forward/efficientnet.py:40
Functionstage_info
(self, stage)
custom_forward/resnet.py:43
Functionstage_info
(self, stage)
custom_forward/vision_transformer.py:40
Functionswin_nano_patch4_window7_224
Swin-M @ 224x224, trained ImageNet-1k
custom_model/swin_transformer.py:8
Functionswin_pico_patch4_window7_224
Swin-M @ 224x224, trained ImageNet-1k
custom_model/swin_transformer.py:17
Functionvip_l7
(pretrained=False, pretrained_cfg=None, **kwargs)
custom_model/vip.py:263
Functionvip_m7
(pretrained=False, pretrained_cfg=None, **kwargs)
custom_model/vip.py:249
Functionvip_s14
(pretrained=False, pretrained_cfg=None, **kwargs)
custom_model/vip.py:221
← previousnext →101–200 of 201, ranked by callers