MCPcopy Create free account

hub / github.com/Hao840/OFAKD / functions

Functions201 in github.com/Hao840/OFAKD

⨍ Functions 201◇ Types & classes 48
↓ 17 callersMethodupdate
(self)
utils.py:111
↓ 14 callersMethodcuda
(self)
distillers/crd.py:231
↓ 7 callersMethod__init__
(self)
distillers/utils.py:46
↓ 6 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
custom_model/cycle_mlp.py:36
↓ 5 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
custom_model/vip.py:28
↓ 5 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, drop=0.)
custom_model/hire_mlp.py:26
↓ 4 callersMethod__init__
(self, student, teacher, criterion, args, **kwargs)
distillers/crd.py:14
↓ 4 callersFunction_cfg
(url='', **kwargs)
custom_model/hire_mlp.py:15
↓ 4 callersMethodforward_embeddings
(self, x)
custom_model/cycle_mlp.py:317
↓ 3 callersFunction_cfg
(url='', **kwargs)
custom_model/vip.py:10
↓ 3 callersFunction_cfg
(url='', **kwargs)
custom_model/cycle_mlp.py:18
↓ 3 callersFunctionis_cnn_model
(distiller)
distillers/utils.py:169
↓ 2 callersMethod__init__
(self, steps, most_recent=30, drop_first=True)
utils.py:102
↓ 2 callersFunction_pdist
(e, squared, eps)
distillers/rkd.py:60
↓ 2 callersFunctioncat_mask
(t, mask1, mask2)
distillers/dkd.py:74
↓ 2 callersFunctionget_module_dict
(module_dict, k)
distillers/utils.py:188
↓ 2 callersFunctioninter_class_relation
(y_s, y_t)
distillers/dist.py:48
↓ 2 callersFunctionofa_loss
(logits_student, logits_teacher, target_mask, eps, temperature=1.)
distillers/ofa.py:13
↓ 2 callersFunctionregister_new_forward
(model)
custom_forward/registry.py:17
↓ 2 callersFunctionset_module_dict
(module_dict, k, v)
distillers/utils.py:182
↓ 2 callersFunctionvalidate
(args)
validate.py:128
↓ 2 callersFunctionvalidate
(model, loader, loss_fn, args, amp_autocast=suppress, log_suffix='')
train.py:888
↓ 1 callersMethod__getitem__
Args: index (int): Index Returns: tuple: (image, target) where target is class_index of the target class.
utils.py:41
↓ 1 callersMethod__init__
(self, student, teacher, criterion, args, **kwargs)
distillers/_base.py:36
↓ 1 callersMethod__init__
(self, student, teacher, criterion, args, **kwargs)
distillers/correlation.py:12
↓ 1 callersMethod__init__
(self, student, teacher, criterion, args, **kwargs)
distillers/kd.py:13
↓ 1 callersFunction_get_gt_mask
(logits, target)
distillers/dkd.py:62
↓ 1 callersFunction_get_other_mask
(logits, target)
distillers/dkd.py:68
↓ 1 callersFunction_parse_args
()
train.py:351
↓ 1 callersFunction_try_run
(args, initial_batch_size)
validate.py:309
↓ 1 callersFunction_unpatchify
x: (N, L, patch_size**2 *C) imgs: (N, C, H, W)
distillers/utils.py:218
↓ 1 callersFunctionbasic_blocks
(dim, index, layers, segment_dim, mlp_ratio=3., qkv_bias=False, qk_scale=None, \ attn_drop=0
custom_model/vip.py:133
↓ 1 callersFunctionbasic_blocks
(dim, index, layers, mlp_ratio=3., qkv_bias=False, qk_scale=None, attn_drop=0., drop_path_ra
custom_model/cycle_mlp.py:236
↓ 1 callersFunctionbasic_blocks
(dim, index, layers, mlp_ratio=4., attn_drop=0., drop_path_rate=0., pixel=2, step_stride=2,
custom_model/hire_mlp.py:219
↓ 1 callersFunctioncosine_similarity
(a, b, eps=1e-8)
distillers/dist.py:39
↓ 1 callersFunctiondist_loss
(logits_student, logits_teacher, beta=1., gamma=1., temperature=1.)
distillers/dist.py:31
↓ 1 callersFunctiondkd_loss
(logits_student, logits_teacher, target, alpha, beta, temperature)
distillers/dkd.py:37
↓ 1 callersMethoddraw
Draw N samples from multinomial
distillers/crd.py:235
↓ 1 callersMethodforward_embeddings
(self, x)
custom_model/vip.py:199
↓ 1 callersMethodforward_embeddings
(self, x)
custom_model/hire_mlp.py:281
↓ 1 callersMethodforward_tokens
(self, x)
custom_model/vip.py:205
↓ 1 callersMethodforward_tokens
(self, x)
custom_model/cycle_mlp.py:323
↓ 1 callersMethodforward_tokens
(self, x)
custom_model/hire_mlp.py:285
↓ 1 callersMethodgen_offset
offset (Tensor[batch_size, 2 * offset_groups * kernel_height * kernel_width, out_height, out_width]): offsets to be applied for
custom_model/cycle_mlp.py:106
↓ 1 callersFunctionget_distiller
(name)
distillers/registry.py:13
↓ 1 callersMethodget_learnable_parameters
(self)
distillers/_base.py:17
↓ 1 callersMethodget_pred_text
(self)
utils.py:122
↓ 1 callersFunctionintra_class_relation
(y_s, y_t)
distillers/dist.py:52
↓ 1 callersFunctionkd_loss
(logits_student, logits_teacher, temperature=1.)
distillers/utils.py:161
↓ 1 callersFunctionmain
()
validate.py:330
↓ 1 callersFunctionmain
()
train.py:368
↓ 1 callersFunctionpearson_correlation
(a, b, eps=1e-8)
distillers/dist.py:43
↓ 1 callersMethodreset_parameters
(self)
custom_model/cycle_mlp.py:98
↓ 1 callersFunctiontrain_one_epoch
( epoch, distiller, loader, optimizer, args, lr_scheduler=None, saver=None, output_dir=None, a
train.py:761
↓ 1 callersFunctionwrite_results
(results_file, results)
validate.py:383
FunctionCycleMLP_B1
(pretrained=False, pretrained_cfg=None, **kwargs)
custom_model/cycle_mlp.py:351
FunctionCycleMLP_B2
(pretrained=False, pretrained_cfg=None, **kwargs)
custom_model/cycle_mlp.py:364
FunctionCycleMLP_B3
(pretrained=False, pretrained_cfg=None, **kwargs)
custom_model/cycle_mlp.py:377
FunctionCycleMLP_B4
(pretrained=False, pretrained_cfg=None, **kwargs)
custom_model/cycle_mlp.py:390
FunctionCycleMLP_B5
(pretrained=False, pretrained_cfg=None, **kwargs)
custom_model/cycle_mlp.py:403
Method__getitem__
(self, index)
utils.py:88
Method__init__
(self, root, name, class_map, load_bytes, is_sample=False, k=4096, **kwargs)
utils.py:13
Method__init__
(self, root, train, is_sample=False, k=4096, **kwargs)
utils.py:64
Method__init__
(self, student, teacher, criterion, args, **kwargs)
distillers/fitnet.py:14
Method__init__
(self, power=2)
distillers/crd.py:54
Method__init__
(self, dim_in=1024, dim_out=128)
distillers/crd.py:67
Method__init__
(self, num_data)
distillers/crd.py:82
Method__init__
(self, inputSize, output_size, K, T=0.07, momentum=0.5)
distillers/crd.py:110
Method__init__
(self, probs)
distillers/crd.py:194
Method__init__
(self, student, teacher, criterion, args)
distillers/_base.py:7
Method__init__
(self, student, teacher, criterion, args, **kwargs)
distillers/ofa.py:25
Method__init__
(self, student, teacher, criterion, args, **kwargs)
distillers/rkd.py:12
Method__init__
(self, input_resolution, dim, out_dim=None, norm_layer=nn.LayerNorm, act_layer=nn.Identity)
distillers/utils.py:9
Method__init__
(self, number=1, inverse=False, remove_mode=True)
distillers/utils.py:56
Method__init__
(self, token_num=0, fn: nn.Module = nn.Identity(), token_fn: nn.Module = nn.Identity(), inverse=False)
distillers/utils.py:75
Method__init__
(self, lambda_fn)
distillers/utils.py:100
Method__init__
(self, out_patch_num)
distillers/utils.py:109
Method__init__
(self, channel_in, channel_out, kernel_size=3, stride=2, padding=1, affine=True)
distillers/utils.py:128
Method__init__
(self, p)
distillers/utils.py:210
Method__init__
(self, dim_in=1024, dim_out=128)
distillers/correlation.py:40
Method__init__
(self, student, teacher, criterion, args, **kwargs)
distillers/dist.py:11
Method__init__
(self, student, teacher, criterion, args, **kwargs)
distillers/kd.py:36
Method__init__
(self, student, teacher, criterion, args, **kwargs)
distillers/dkd.py:12
Method__init__
(self, dim, segment_dim=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0.)
custom_model/vip.py:47
Method__init__
(self, dim, segment_dim, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0.,
custom_model/vip.py:85
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768)
custom_model/vip.py:109
Method__init__
(self, in_embed_dim, out_embed_dim, patch_size)
custom_model/vip.py:122
Method__init__
(self, layers, img_size=224, patch_size=4, in_chans=3, num_classes=1000, embed_dims=None, tr
custom_model/vip.py:151
Method__init__
( self, in_channels: int, out_channels: int, kernel_size,
custom_model/cycle_mlp.py:58
Method__init__
(self, dim, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0.)
custom_model/cycle_mlp.py:148
Method__init__
(self, dim, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0., drop_path=0.
custom_model/cycle_mlp.py:179
Method__init__
(self, patch_size=16, stride=16, padding=0, in_chans=3, embed_dim=768, norm_layer=None, groups=1)
custom_model/cycle_mlp.py:203
Method__init__
(self, in_embed_dim, out_embed_dim, patch_size)
custom_model/cycle_mlp.py:224
Method__init__
(self, layers, img_size=224, patch_size=4, in_chans=3, num_classes=1000, embed_dims=None, tr
custom_model/cycle_mlp.py:252
Method__init__
(self, dim, attn_drop=0., proj_drop=0., pixel=2, step=1, step_pad_mode='c', pixel_pad_mode='
custom_model/hire_mlp.py:45
Method__init__
(self, dim, mlp_ratio=4., drop=0., attn_drop=0., drop_path=0., pixel=2, step=1, step_pad_mod
custom_model/hire_mlp.py:158
Method__init__
(self, patch_size=16, stride=16, padding=0, in_chans=3, embed_dim=768, groups=1)
custom_model/hire_mlp.py:182
Method__init__
(self, in_embed_dim, out_embed_dim, patch_size)
custom_model/hire_mlp.py:205
Method__init__
(self, layers, img_size=224, patch_size=4, in_chans=3, num_classes=1000, embed_dims=None, ml
custom_model/hire_mlp.py:236
Method__init__
(self, num_classes=1000)
custom_model/mobilenetv1.py:10
next →1–100 of 201, ranked by callers