MCPcopy Create free account

hub / github.com/NUS-HPC-AI-Lab/InfoBatch / functions

Functions105 in github.com/NUS-HPC-AI-Lab/InfoBatch

⨍ Functions 105◇ Types & classes 16
↓ 14 callersFunctionsafe_print
(*args, **kwargs)
examples/cifar_example.py:21
↓ 4 callersMethod_make_layer
(self, block, planes, num_blocks, stride)
research/model.py:86
↓ 4 callersMethod_make_layer
(self, block, planes, num_blocks, stride)
examples/model.py:86
↓ 4 callersMethodreset
(self)
infobatch/infobatch.py:179
↓ 4 callersMethodstep
Performs a single optimization step. Arguments: closure (callable, optional): A closure that reevaluates the model
examples/lars.py:73
↓ 3 callersMethodupdate
(self, values)
infobatch/infobatch.py:92
↓ 2 callersFunctionResNet18
(num_classes=10)
research/model.py:117
↓ 2 callersFunctionResNet18
(num_classes=10)
examples/model.py:117
↓ 2 callersFunctionResNet50
(num_classes=10)
research/model.py:125
↓ 2 callersFunctionResNet50
(num_classes=10)
examples/model.py:125
↓ 2 callersMethod__init__
(self, block, num_blocks, num_classes=10)
research/model.py:72
↓ 2 callersMethod__init__
(self, block, num_blocks, num_classes=10)
examples/model.py:72
↓ 2 callersMethodget_pruned_count
(self)
infobatch/infobatch.py:156
↓ 2 callersMethodreset
(self)
research/infobatch.py:79
↓ 2 callersMethodreset_weights
(self)
research/infobatch.py:66
↓ 2 callersMethodreset_weights
(self)
infobatch/infobatch.py:163
↓ 2 callersMethodstep
Performs a single optimization step. Arguments: closure (callable, optional): A closure that reevaluates the model
research/lars.py:73
↓ 2 callersMethodtotal_save
(self)
research/infobatch.py:63
↓ 1 callersFunctionResNet101
(num_classes=10)
research/model.py:129
↓ 1 callersFunctionResNet101
(num_classes=10)
examples/model.py:129
↓ 1 callersMethod__init__
(self, dataset: Dataset, num_epochs: int, prune_ratio: float = 0.5, delta: float = 0.875)
infobatch/infobatch.py:73
↓ 1 callersMethod__setscore__
(self, indices, values)
research/infobatch.py:16
↓ 1 callersFunctionconcat_all_gather
Performs all_gather operation on the provided tensors. *** Warning ***: torch.distributed.all_gather has no gradient.
infobatch/infobatch.py:45
↓ 1 callersFunctiondestroy_ddp
()
examples/cifar_example.py:30
↓ 1 callersFunctionget_bias_parameters
(module, exclude_func=None)
research/lars.py:148
↓ 1 callersFunctionget_bias_parameters
(module, exclude_func=None)
examples/lars.py:148
↓ 1 callersFunctionget_common_parameters
(module, exclude_func=None)
research/lars.py:165
↓ 1 callersFunctionget_common_parameters
(module, exclude_func=None)
examples/lars.py:165
↓ 1 callersFunctionget_norm_bias_parameters
(module)
research/lars.py:158
↓ 1 callersFunctionget_norm_bias_parameters
(module)
examples/lars.py:158
↓ 1 callersFunctionget_norm_parameters
(module)
research/lars.py:144
↓ 1 callersFunctionget_norm_parameters
(module)
examples/lars.py:144
↓ 1 callersFunctionget_parameters_from_cls
(module, cls_)
research/lars.py:132
↓ 1 callersFunctionget_parameters_from_cls
(module, cls_)
examples/lars.py:132
↓ 1 callersMethodno_prune
(self)
research/infobatch.py:49
↓ 1 callersMethodno_prune
(self)
infobatch/infobatch.py:145
↓ 1 callersMethodprune
(self)
research/infobatch.py:27
↓ 1 callersMethodprune
(self)
infobatch/infobatch.py:119
↓ 1 callersMethodpruning_sampler
(self)
research/infobatch.py:46
↓ 1 callersMethodset_active_indices
(self, cur_batch_indices: torch.Tensor)
infobatch/infobatch.py:89
↓ 1 callersFunctionsetup_ddp
()
examples/cifar_example.py:25
↓ 1 callersFunctiontest
(epoch)
research/cifar_example.py:145
↓ 1 callersFunctiontest
(epoch)
examples/cifar_example.py:227
↓ 1 callersFunctiontrain
(epoch)
research/cifar_example.py:116
↓ 1 callersFunctiontrain_info_batch
(epoch)
examples/cifar_example.py:172
↓ 1 callersFunctiontrain_normal
(epoch)
examples/cifar_example.py:201
FunctionResNet152
()
research/model.py:133
FunctionResNet152
()
examples/model.py:133
FunctionResNet34
()
research/model.py:121
FunctionResNet34
()
examples/model.py:121
Method__getattr__
(self, name)
infobatch/infobatch.py:85
Method__getitem__
(self, index)
research/infobatch.py:22
Method__getitem__
(self, index)
infobatch/infobatch.py:114
Method__getitem__
(self, idx)
infobatch/infobatch.py:176
Method__getitem__
Gets element of the dataset. Args: index: index of the element in the dataset Returns: Single
infobatch/infobatch.py:227
Method__init__
( self, params: Iterable[torch.nn.Parameter], lr=1e-3, momentu
research/lars.py:44
Method__init__
(self, params, lr=1e-3, betas=(0.9, 0.999), eps=1e-6, weight_decay=0, adam=False)
research/lamb.py:45
Method__init__
(self, dataset, ratio = 0.5, num_epoch=None, delta = 0.875)
research/infobatch.py:6
Method__init__
(self, infobatch_dataset, num_epoch = math.inf, delta = 1)
research/infobatch.py:72
Method__init__
(self, in_planes, planes, stride=1)
research/model.py:15
Method__init__
(self, in_planes, planes, stride=1)
research/model.py:43
Method__init__
(self, dataset: InfoBatch)
infobatch/infobatch.py:168
Method__init__
(self, sampler: IBSampler)
infobatch/infobatch.py:216
Method__init__
(self, dataset: IBSampler, num_replicas: Optional[int] = None, rank: Optional[int] = None, sh
infobatch/infobatch.py:238
Method__init__
( self, params: Iterable[torch.nn.Parameter], lr=1e-3, momentu
examples/lars.py:44
Method__init__
(self, params, lr=1e-3, betas=(0.9, 0.999), eps=1e-6, weight_decay=0, adam=False)
examples/lamb.py:45
Method__init__
(self, in_planes, planes, stride=1)
examples/model.py:15
Method__init__
(self, in_planes, planes, stride=1)
examples/model.py:43
Method__iter__
(self)
research/infobatch.py:102
Method__iter__
(self)
infobatch/infobatch.py:198
Method__iter__
Notes self.dataset is actually an instance of IBSampler rather than InfoBatch.
infobatch/infobatch.py:248
Method__len__
(self)
research/infobatch.py:19
Method__len__
(self)
research/infobatch.py:99
Method__len__
(self)
infobatch/infobatch.py:111
Method__len__
(self)
infobatch/infobatch.py:195
Method__len__
(self)
infobatch/infobatch.py:224
Method__next__
(self)
research/infobatch.py:91
Method__next__
(self)
infobatch/infobatch.py:192
Methodcreate_emb
(self,x)
research/model.py:106
Methodcreate_emb
(self,x)
examples/model.py:106
Functioncreate_optimizer_lars
(model, lr, momentum, weight_decay, bn_bias_separately, epsilon)
research/lars.py:14
Functioncreate_optimizer_lars
(model, lr, momentum, weight_decay, bn_bias_separately, epsilon)
examples/lars.py:14
Methodforward
(self, x)
research/model.py:32
Methodforward
(self, x)
research/model.py:62
Methodforward
(self, x)
research/model.py:94
Methodforward
(self, x)
examples/model.py:32
Methodforward
(self, x)
examples/model.py:62
Methodforward
(self, x)
examples/model.py:94
Functionget_members_fn
(m)
research/lars.py:133
Functionget_members_fn
(m)
examples/lars.py:133
Methodget_weights
(self,indexes)
research/infobatch.py:60
Methodget_weights
(self, indexes)
infobatch/infobatch.py:153
Functioninfo_hack_indices
(self)
infobatch/infobatch.py:15
Methodmean_score
(self)
research/infobatch.py:54
Methodmean_score
(self)
infobatch/infobatch.py:150
Methodnormal_sampler_no_prune
(self)
research/infobatch.py:57
Methodreset
(self, )
infobatch/infobatch.py:220
Methodsampler
(self)
infobatch/infobatch.py:139
Methodset_decay
(self,weight_decay)
research/lars.py:68
Methodset_decay
(self,weight_decay)
examples/lars.py:68
next →1–100 of 105, ranked by callers