Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/SLDGroup/MERIT
/ functions
Functions
932 in github.com/SLDGroup/MERIT
⨍
Functions
932
◇
Types & classes
209
↓ 1 callers
Function
grid_reverse_nchw
(windows, grid_size: List[int], img_size: List[int])
lib/maxxvit_4out.py:1333
↓ 1 callers
Function
grid_reverse_nchw
(windows, grid_size: List[int], img_size: List[int])
lib/models_timm/maxxvit.py:1332
↓ 1 callers
Function
hard_mish
Hard Mish Experimental, based on notes by Mish author Diganta Misra at https://github.com/digantamisra98/H-Mish/blob/0da20d4bc58e696b6803f2
lib/models_timm/layers/activations.py:104
↓ 1 callers
Function
hard_mish_jit
Hard Mish Experimental, based on notes by Mish author Diganta Misra at https://github.com/digantamisra98/H-Mish/blob/0da20d4bc58e696b6803f2
lib/models_timm/layers/activations_jit.py:77
↓ 1 callers
Function
hard_mish_jit_bwd
(x, grad_output)
lib/models_timm/layers/activations_me.py:184
↓ 1 callers
Function
hard_mish_jit_fwd
(x)
lib/models_timm/layers/activations_me.py:179
↓ 1 callers
Function
hard_sigmoid
(x, inplace: bool = False)
lib/models_timm/layers/activations.py:88
↓ 1 callers
Function
hard_sigmoid_jit
(x, inplace: bool = False)
lib/models_timm/layers/activations_jit.py:49
↓ 1 callers
Function
hard_sigmoid_jit_bwd
(x, grad_output)
lib/models_timm/layers/activations_me.py:105
↓ 1 callers
Function
hard_sigmoid_jit_fwd
(x, inplace: bool = False)
lib/models_timm/layers/activations_me.py:100
↓ 1 callers
Function
hard_swish
(x, inplace: bool = False)
lib/models_timm/layers/activations.py:74
↓ 1 callers
Function
hard_swish_jit
(x, inplace: bool = False)
lib/models_timm/layers/activations_jit.py:63
↓ 1 callers
Function
hard_swish_jit_bwd
(x, grad_output)
lib/models_timm/layers/activations_me.py:140
↓ 1 callers
Function
hard_swish_jit_fwd
(x)
lib/models_timm/layers/activations_me.py:135
↓ 1 callers
Function
hf_split
(hf_id)
lib/models_timm/hub.py:65
↓ 1 callers
Function
inference
(args, model, test_save_path=None)
test_synapse.py:49
↓ 1 callers
Function
inference
(args, model, best_performance)
trainer.py:26
↓ 1 callers
Method
init_weights
(self)
lib/models_timm/vision_transformer_relpos.py:224
↓ 1 callers
Method
init_weights
(self)
lib/models_timm/vision_transformer_relpos.py:323
↓ 1 callers
Method
init_weights
(self, mode='')
lib/models_timm/vision_transformer_relpos.py:449
↓ 1 callers
Function
instance_std
(x, eps: float = 1e-5)
lib/models_timm/layers/evo_norm.py:36
↓ 1 callers
Function
is_model
Check if a model name exists
lib/models_timm/registry.py:99
↓ 1 callers
Function
is_static_pad
(kernel_size: int, stride: int = 1, dilation: int = 1, **_)
lib/models_timm/layers/padding.py:23
↓ 1 callers
Function
load_cfg_from_json
(json_file: Union[str, os.PathLike])
lib/models_timm/hub.py:74
↓ 1 callers
Function
load_checkpoint
(model, checkpoint_path, use_ema=True, strict=True, remap=False)
lib/models_timm/helpers.py:66
↓ 1 callers
Function
load_custom_pretrained
r"""Loads a custom (read non .pth) weight file Downloads checkpoint file into cache-dir like torch.hub based loaders, but calls a passed in c
lib/models_timm/helpers.py:177
↓ 1 callers
Function
load_model_config_from_hf
(model_id: str)
lib/models_timm/hub.py:85
↓ 1 callers
Function
load_pretrained
Load pretrained checkpoint Args: model (nn.Module) : PyTorch model module pretrained_cfg (Optional[Dict]): configuration for pre
lib/models_timm/helpers.py:240
↓ 1 callers
Function
make_safe
(name)
lib/models_timm/factory.py:23
↓ 1 callers
Function
mish
Mish: A Self Regularized Non-Monotonic Neural Activation Function - https://arxiv.org/abs/1908.08681 NOTE: I don't have a working inplace variant
lib/models_timm/layers/activations.py:29
↓ 1 callers
Function
mish_jit
Mish: A Self Regularized Non-Monotonic Neural Activation Function - https://arxiv.org/abs/1908.08681
lib/models_timm/layers/activations_jit.py:26
↓ 1 callers
Function
mish_jit_bwd
(x, grad_output)
lib/models_timm/layers/activations_me.py:66
↓ 1 callers
Function
mish_jit_fwd
(x)
lib/models_timm/layers/activations_me.py:61
↓ 1 callers
Function
model_entrypoint
Fetch a model entrypoint for specified model name
lib/models_timm/registry.py:105
↓ 1 callers
Method
module_name
feature module name accessor
lib/models_timm/features.py:72
↓ 1 callers
Function
named_modules_with_params
(module: nn.Module, name='', depth_first=True, include_root=False)
lib/models_timm/helpers.py:627
↓ 1 callers
Function
num_groups
(group_size, channels)
lib/maxxvit_4out.py:950
↓ 1 callers
Function
num_groups
(group_size, channels)
lib/models_timm/maxxvit.py:949
↓ 1 callers
Function
pretrained_cfg_for_features
(pretrained_cfg)
lib/models_timm/helpers.py:428
↓ 1 callers
Function
random_rot_flip
(image, label)
utils/dataset_ACDC.py:13
↓ 1 callers
Function
random_rot_flip
(image, label)
utils/dataset_synapse.py:12
↓ 1 callers
Function
random_rotate
(image, label)
utils/dataset_ACDC.py:23
↓ 1 callers
Function
random_rotate
(image, label)
utils/dataset_synapse.py:22
↓ 1 callers
Function
rel_pos_indices
(size)
lib/models_timm/layers/lambda_layer.py:31
↓ 1 callers
Function
remap_checkpoint
remap checkpoint by iterating over state dicts in order (ignoring original keys). This assumes models (and originating state dict) were created w
lib/models_timm/helpers.py:81
↓ 1 callers
Method
reset
(self)
utils/utils.py:57
↓ 1 callers
Method
reset_parameters
(self)
lib/models_timm/layers/non_local_attn.py:57
↓ 1 callers
Method
reset_parameters
(self)
lib/models_timm/layers/bottleneck_attn.py:130
↓ 1 callers
Method
reset_parameters
(self)
lib/models_timm/layers/inplace_abn.py:74
↓ 1 callers
Method
reset_parameters
(self)
lib/models_timm/layers/global_context.py:43
↓ 1 callers
Method
reset_parameters
(self)
lib/models_timm/layers/lambda_layer.py:102
↓ 1 callers
Method
reset_parameters
(self)
lib/models_timm/layers/evo_norm.py:111
↓ 1 callers
Method
reset_parameters
(self)
lib/models_timm/layers/evo_norm.py:149
↓ 1 callers
Method
reset_parameters
(self)
lib/models_timm/layers/evo_norm.py:184
↓ 1 callers
Method
reset_parameters
(self)
lib/models_timm/layers/evo_norm.py:223
↓ 1 callers
Method
reset_parameters
(self)
lib/models_timm/layers/evo_norm.py:278
↓ 1 callers
Method
reset_parameters
(self)
lib/models_timm/layers/evo_norm.py:327
↓ 1 callers
Method
reset_parameters
(self)
lib/models_timm/layers/halo_attn.py:162
↓ 1 callers
Method
reset_parameters
(self)
lib/models_timm/layers/cond_conv2d.py:73
↓ 1 callers
Method
reset_parameters
(self)
lib/models_timm/layers/filter_response_norm.py:30
↓ 1 callers
Method
reset_parameters
(self)
lib/models_timm/layers/filter_response_norm.py:58
↓ 1 callers
Function
resolve_pretrained_cfg
(variant: str, pretrained_cfg=None)
lib/models_timm/helpers.py:482
↓ 1 callers
Function
save_for_hf
(model, save_directory, model_config=None)
lib/models_timm/hub.py:102
↓ 1 callers
Function
set_default_kwargs
(kwargs, names, pretrained_cfg)
lib/models_timm/helpers.py:437
↓ 1 callers
Function
stem_b16
(in_chs, out_chs, activation, resolution=224)
lib/models_timm/levit.py:201
↓ 1 callers
Function
swish
Swish - Described in: https://arxiv.org/abs/1710.05941
lib/models_timm/layers/activations.py:14
↓ 1 callers
Function
swish_jit
Swish - Described in: https://arxiv.org/abs/1710.05941
lib/models_timm/layers/activations_jit.py:19
↓ 1 callers
Function
swish_jit_bwd
(x, grad_output)
lib/models_timm/layers/activations_me.py:23
↓ 1 callers
Function
swish_jit_fwd
(x)
lib/models_timm/layers/activations_me.py:18
↓ 1 callers
Function
update_pretrained_cfg_and_kwargs
Update the default_cfg and kwargs before passing to model Args: pretrained_cfg: input pretrained cfg (updated in-place) kwargs:
lib/models_timm/helpers.py:464
↓ 1 callers
Function
val
()
train_ACDC.py:122
↓ 1 callers
Function
val_single_volume
(image, label, net, classes, patch_size=[256, 256], test_save_path=None, case=None, z_spacing=1)
utils/utils.py:249
↓ 1 callers
Function
variance_scaling_
(tensor, scale=1.0, mode='fan_in', distribution='normal')
lib/models_timm/layers/weight_init.py:99
↓ 1 callers
Function
window_partition_nchw
(x, window_size: List[int])
lib/maxxvit_4out.py:1305
↓ 1 callers
Function
window_partition_nchw
(x, window_size: List[int])
lib/models_timm/maxxvit.py:1304
↓ 1 callers
Function
window_reverse_nchw
(windows, window_size: List[int], img_size: List[int])
lib/maxxvit_4out.py:1315
↓ 1 callers
Function
window_reverse_nchw
(windows, window_size: List[int], img_size: List[int])
lib/models_timm/maxxvit.py:1314
Function
CalParams
Usage: Calculate Params and FLOPs via [THOP](https://github.com/Lyken17/pytorch-OpCounter) Necessarity: from thop import prof
utils/utils.py:75
Method
__call__
(self, sample)
utils/dataset_ACDC.py:34
Method
__call__
(self, sample)
utils/dataset_synapse.py:33
Method
__call__
(self, h: torch.Tensor, duplicate_pooling: torch.Tensor, out_extrap: torch.Tensor)
lib/models_timm/layers/ml_decoder.py:96
Method
__call__
(self, x: torch.Tensor)
lib/models_timm/layers/space_to_depth.py:21
Method
__enter__
(self)
lib/models_timm/layers/config.py:35
Method
__enter__
(self)
lib/models_timm/layers/config.py:54
Method
__enter__
(self)
lib/models_timm/layers/config.py:73
Method
__enter__
(self)
lib/models_timm/layers/config.py:106
Method
__exit__
(self, *args: Any)
lib/models_timm/layers/config.py:38
Method
__exit__
(self, *args: Any)
lib/models_timm/layers/config.py:57
Method
__exit__
(self, *args: Any)
lib/models_timm/layers/config.py:76
Method
__exit__
(self, *args: Any)
lib/models_timm/layers/config.py:109
Method
__getitem__
(self, idx)
utils/dataset_ACDC.py:61
Method
__getitem__
(self, idx)
utils/dataset_synapse.py:61
Method
__getitem__
(self, item)
lib/models_timm/features.py:77
Method
__init__
(self, output_size)
utils/dataset_ACDC.py:31
Method
__init__
(self, base_dir, list_dir, split, transform=None)
utils/dataset_ACDC.py:52
Method
__init__
(self, output_size)
utils/dataset_synapse.py:30
Method
__init__
(self, base_dir, list_dir, split, nclass=9, transform=None)
utils/dataset_synapse.py:51
Method
__init__
(self, n_classes)
utils/utils.py:107
Method
__init__
( self, dim: int, dim_out: Optional[int] = None, dim_head: int
lib/maxxvit_4out.py:689
Method
__init__
( self, dim: int, dim_out: Optional[int] = None, dim_head: int
lib/maxxvit_4out.py:734
← previous
next →
201–300 of 932, ranked by callers