MCPcopy Create free account

hub / github.com/bic-L/MaxFormer / functions

Functions449 in github.com/bic-L/MaxFormer

↓ 57 callersFunctionprint
(*args, **kwargs)
imagenet/misc.py:177
↓ 34 callersFunctionprint
(*args, **kwargs)
event/utils.py:195
↓ 18 callersMethod__init__
(self, dim, num_heads=8)
imagenet/mixer_hub.py:214
↓ 18 callersMethod__init__
(self, dim, num_heads=8)
cifar10-100/mixer_hub.py:207
↓ 18 callersMethod__init__
(self, dim, num_heads=8)
event/mixer_hub.py:206
↓ 18 callersMethodmax
(self)
event/utils.py:57
↓ 17 callersMethodupdate
(self, **kwargs)
event/utils.py:78
↓ 8 callersMethod__init__
(self, in_channels=2, out_channels=256, kernel_size = 3, stride = 1, padding = 1, shortcut= False)
imagenet/embedding_hub.py:7
↓ 8 callersMethod__init__
(self, in_channels=2, out_channels=256, kernel_size = 3, stride = 1, padding = 1, shortcut= False)
cifar10-100/embedding_hub.py:6
↓ 8 callersMethod__init__
(self, in_channels=2, out_channels=256, kernel_size = 3, stride = 1, padding = 1, shortcut= False)
event/embedding_hub.py:5
↓ 7 callersFunction_check_args_tf
(kwargs)
cifar10-100/aa_snn.py:50
↓ 7 callersMethodload_state_dict
(self, state_dict)
imagenet/misc.py:277
↓ 7 callersMethodstate_dict
(self)
imagenet/misc.py:274
↓ 7 callersMethodupdate
(self, **kwargs)
imagenet/misc.py:92
↓ 5 callersFunction_randomly_negate
With 50% prob, negate the value
cifar10-100/aa_snn.py:173
↓ 5 callersFunctionsplit_to_train_test_set
:param train_ratio: split the ratio of the origin dataset as the train set :type train_ratio: float :param origin_dataset: the origin dat
event/train.py:171
↓ 4 callersFunction_apply_op
(img: Tensor, op_name: str, magnitude: float, interpolation: InterpolationMode, fill: Optional[L
event/autoaugment.py:81
↓ 4 callersFunctionconv3x3
3x3 convolution with padding
cifar10-100/max_resnet.py:7
↓ 3 callersMethod__init__
( self, policy: AutoAugmentPolicy = AutoAugmentPolicy.IMAGENET, interpolation: Interpo
event/autoaugment.py:150
↓ 3 callersMethod_make_layer
(self, block, planes, blocks, stride=1, dilate=False)
cifar10-100/max_resnet.py:144
↓ 3 callersFunctionbuild_dataset
(is_train, args)
imagenet/utils.py:110
↓ 3 callersFunctionevaluate
(data_loader, model, device)
imagenet/utils.py:240
↓ 3 callersFunctionis_dist_avail_and_initialized
()
imagenet/misc.py:188
↓ 3 callersFunctionis_dist_avail_and_initialized
()
event/utils.py:203
↓ 3 callersMethodrotate
(self, events, theta)
event/augment.py:132
↓ 2 callersMethod__init__
( self, in_channels=2, num_classes=11, embed_dims=512, mlp_ratios=4,
imagenet/ms_qkformer.py:101
↓ 2 callersMethod__init__
(self, block, layers, num_classes=10, groups=1, width_per_group=64, replace_stride_with_dilat
cifar10-100/max_resnet.py:107
↓ 2 callersMethod__len__
(self)
event/dataset.py:141
↓ 2 callersFunctionaccuracy
Computes the accuracy over the k top predictions for the specified values of k
event/utils.py:162
↓ 2 callersMethodadd_meter
(self, name, meter)
event/utils.py:105
↓ 2 callersFunctioncreate_loader
( dataset, input_size, batch_size, is_training=False, use_prefetcher=T
cifar10-100/loader.py:147
↓ 2 callersFunctioncreate_transform
( input_size, is_training=False, use_prefetcher=False, no_aug=False, s
cifar10-100/transforms_factory.py:168
↓ 2 callersFunctionevaluate
(model, criterion, data_loader, device, print_freq=100, header='Test:')
event/train.py:277
↓ 2 callersFunctionget_world_size
()
imagenet/misc.py:196
↓ 2 callersMethodlog_every
(self, iterable, print_freq, header=None)
imagenet/misc.py:124
↓ 2 callersMethodlog_every
(self, iterable, print_freq, header=None)
event/utils.py:108
↓ 2 callersFunctionmax_resnet18
(pretrained=False, progress=True, **kwargs)
cifar10-100/max_resnet.py:197
↓ 2 callersMethodscale
(self, events, factor)
event/augment.py:478
↓ 2 callersFunctionsetup_for_distributed
This function disables printing when not in master process
imagenet/misc.py:171
↓ 2 callersMethodsynchronize_between_processes
(self)
imagenet/misc.py:117
↓ 2 callersMethodsynchronize_between_processes
(self)
event/utils.py:101
↓ 2 callersMethodupdate
(self, value, n=1)
imagenet/misc.py:38
↓ 2 callersMethodupdate
(self, value, n=1)
event/utils.py:24
↓ 2 callersFunctionvalidate
(model, loader, loss_fn, args, amp_autocast=suppress, log_suffix='')
cifar10-100/train.py:746
↓ 1 callersMethod__init__
(self, loader, mean=IMAGENET_DEFAULT_MEAN, std=IMAGENET_DEF
cifar10-100/loader.py:59
↓ 1 callersMethod__init__
(self, dataset, args, device)
event/dataset.py:128
↓ 1 callersMethod__iter__
(self)
cifar10-100/loader.py:81
↓ 1 callersMethod__str__
(self)
event/utils.py:93
↓ 1 callersMethod_apply_basic
(self, img, mixing_weights, m)
cifar10-100/aa_snn.py:784
↓ 1 callersMethod_apply_blended
(self, img, mixing_weights, m)
cifar10-100/aa_snn.py:768
↓ 1 callersMethod_augmentation_space
(self, num_bins: int)
event/autoaugment.py:35
↓ 1 callersMethod_augmentation_space
(self, num_bins: int, image_size: List[int])
event/autoaugment.py:253
↓ 1 callersMethod_augmentation_space
(self, num_bins: int, image_size: List[int])
event/autoaugment.py:345
↓ 1 callersMethod_augmentation_space
(self, num_bins: int)
event/autoaugment.py:424
↓ 1 callersMethod_calc_blended_weights
(self, ws, m)
cifar10-100/aa_snn.py:758
↓ 1 callersMethod_forward_impl
(self, x)
cifar10-100/max_resnet.py:171
↓ 1 callersMethod_get_policies
( self, policy: AutoAugmentPolicy )
event/autoaugment.py:162
↓ 1 callersFunction_interpolation
(kwargs)
cifar10-100/aa_snn.py:42
↓ 1 callersFunction_parse_args
()
cifar10-100/train.py:307
↓ 1 callersFunction_posterize_level_to_arg
(level, _hparams)
cifar10-100/aa_snn.py:220
↓ 1 callersFunction_resnet
(arch, block, layers, **kwargs)
cifar10-100/max_resnet.py:193
↓ 1 callersFunction_select_rand_weights
(weight_idx=0, transforms=None)
cifar10-100/aa_snn.py:625
↓ 1 callersFunction_solarize_level_to_arg
(level, _hparams)
cifar10-100/aa_snn.py:241
↓ 1 callersMethodadd_meter
(self, name, meter)
imagenet/misc.py:121
↓ 1 callersFunctionadjust_learning_rate
Decay the learning rate with half-cycle cosine after warmup
imagenet/utils.py:30
↓ 1 callersFunctionaugment_and_mix_transform
Create AugMix PyTorch transform :param config_str: String defining configuration of random augmentation. Consists of multiple sections separated
cifar10-100/aa_snn.py:818
↓ 1 callersFunctionaugmix_ops
(magnitude=10, hparams=None, transforms=None)
cifar10-100/aa_snn.py:738
↓ 1 callersFunctionauto_augment_policy
(name='v0', hparams=None)
cifar10-100/aa_snn.py:497
↓ 1 callersFunctionauto_augment_policy_original
(hparams)
cifar10-100/aa_snn.py:431
↓ 1 callersFunctionauto_augment_policy_originalr
(hparams)
cifar10-100/aa_snn.py:464
↓ 1 callersFunctionauto_augment_policy_v0
(hparams)
cifar10-100/aa_snn.py:364
↓ 1 callersFunctionauto_augment_policy_v0r
(hparams)
cifar10-100/aa_snn.py:397
↓ 1 callersFunctionauto_augment_transform
Create a AutoAugment transform :param config_str: String defining configuration of auto augmentation. Consists of multiple sections separate
cifar10-100/aa_snn.py:532
↓ 1 callersFunctionbuild_transform
(is_train, args)
imagenet/utils.py:121
↓ 1 callersFunctionconv1x1
1x1 convolution
cifar10-100/max_resnet.py:13
↓ 1 callersMethoddrop_by_area
(self, events, area_ratio)
event/augment.py:185
↓ 1 callersMethoddrop_by_time
(self, events, ratio)
event/augment.py:171
↓ 1 callersMethodforward
(self, events)
event/dataset.py:199
↓ 1 callersMethodforward_features
(self, x)
imagenet/max_former.py:85
↓ 1 callersMethodforward_features
(self, x)
imagenet/ms_qkformer.py:172
↓ 1 callersMethodforward_features
(self, x)
cifar10-100/max_former.py:69
↓ 1 callersMethodforward_features
(self, x)
cifar10-100/ms_qkformer.py:69
↓ 1 callersMethodforward_features
(self, x)
event/max_former.py:71
↓ 1 callersMethodforward_features
(self, x)
event/ms_qkformer.py:71
↓ 1 callersFunctionget_args_parser
()
imagenet/train.py:41
↓ 1 callersFunctionget_args_parser
()
imagenet/test.py:31
↓ 1 callersFunctionget_grad_norm_
(parameters, norm_type: float = 2.0)
imagenet/misc.py:281
↓ 1 callersFunctionget_layer_id
(name, num_layers)
imagenet/utils.py:89
↓ 1 callersMethodget_params
Get parameters for autoaugment transformation Returns: params required by the autoaugment transformation
event/autoaugment.py:273
↓ 1 callersFunctionget_rank
()
imagenet/misc.py:202
↓ 1 callersFunctionget_rank
()
event/utils.py:217
↓ 1 callersFunctionis_main_process
()
imagenet/misc.py:208
↓ 1 callersFunctionis_main_process
()
event/utils.py:223
↓ 1 callersFunctionload_data
(dataset, dataset_dir, distributed, T)
event/train.py:303
↓ 1 callersFunctionmain
(args)
imagenet/train.py:197
↓ 1 callersFunctionmain
(args)
imagenet/test.py:190
↓ 1 callersFunctionmain
()
cifar10-100/train.py:325
↓ 1 callersFunctionmain
(args)
event/train.py:331
↓ 1 callersMethodmax
(self)
imagenet/misc.py:71
↓ 1 callersFunctionparam_groups_lrd
(model, weight_decay=0.05, no_weight_decay_list=[], layer_decay=.75)
imagenet/utils.py:44
next →1–100 of 449, ranked by callers