MCPcopy Create free account

hub / github.com/DanielShalam/BPA / functions

Functions184 in github.com/DanielShalam/BPA

⨍ Functions 184◇ Types & classes 38
↓ 10 callersMethodcuda
(self)
methods/pt_map/pt_map_loss.py:31
↓ 8 callersMethodcuda
(self)
methods/pt_map/evaluation/test_standard.py:64
↓ 8 callersMethodmax
(self)
utils.py:324
↓ 7 callersMethodupdate
(self, **kwargs)
utils.py:345
↓ 6 callersFunctionmixup_data
Compute the mixup data. Return mixed inputs, pairs of targets, and lambda
models/res_mixup_model.py:21
↓ 5 callersMethodcuda
(self)
methods/pt_map/test_standard.py:63
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
models/res_mixup_model.py:126
↓ 4 callersMethod_make_layer
(self, block, planes, stride=1, drop_rate=0.0, drop_block=False, block_size=1)
models/resnet12.py:97
↓ 4 callersMethodgetAccuracy
(self, probas)
methods/pt_map/evaluation/test_standard.py:130
↓ 4 callersMethodgetProbas
(self)
methods/pt_map/test_standard.py:97
↓ 4 callersMethodgetProbas
(self)
methods/pt_map/evaluation/test_standard.py:96
↓ 4 callersFunctionmixup_data
Compute the mixup data. Return mixed inputs, pairs of targets, and lambda
models/wrn_mixup_model.py:104
↓ 3 callersFunctionC
Modified cost for logarithmic updates
bpa/ot.py:56
↓ 3 callersMethod__init__
(self, in_planes, out_planes, stride, dropRate=0.0)
models/wrn_mixup_model.py:19
↓ 3 callersFunctionconv3x3
3x3 convolution with padding
models/resnet12.py:11
↓ 3 callersFunctioneval_one_epoch
(model, dataloader, fewshot_method, criterion, labels, epoch, args, set_name)
train.py:247
↓ 3 callersMethodgetAccuracy
(self, probas)
methods/pt_map/test_standard.py:133
↓ 2 callersFunctionGenerateRun
(iRun, cfg, regenRState=False, generate=True)
methods/pt_map/FSLTask.py:77
↓ 2 callersMethod__init__
(self, block, layers, num_classes=200, zero_init_residual=False)
models/res_mixup_model.py:94
↓ 2 callersFunctioncenterDatas
(datas)
methods/pt_map/evaluation/test_standard.py:27
↓ 2 callersMethodclone
(self)
methods/pt_map/test_standard.py:58
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
models/res_mixup_model.py:10
↓ 2 callersMethodget_probas
(self, X: torch.Tensor, labels: torch.Tensor)
methods/pt_map/pt_map_loss.py:61
↓ 2 callersMethodlog_every
(self, iterable, print_freq, header=None)
utils.py:371
↓ 2 callersFunctionlog_sum_exp
(u: torch.Tensor, dim: int)
bpa/ot.py:4
↓ 2 callersMethodmask_diagonal
Fill the diagonal of a given matrix (or a batch of them) with given value
bpa/balanced_pairwise_affinities.py:39
↓ 2 callersFunctionparse_args
(script)
methods/pt_map/io_utils.py:7
↓ 2 callersFunctionscaleEachUnitaryDatas
(datas)
methods/pt_map/test_standard.py:34
↓ 2 callersFunctionscaleEachUnitaryDatas
(datas)
methods/pt_map/evaluation/test_standard.py:35
↓ 2 callersMethodupdate
(self, value, n=1)
utils.py:304
↓ 1 callersFunctionQRreduction
(datas)
methods/pt_map/test_standard.py:39
↓ 1 callersFunctionQRreduction
(datas)
methods/pt_map/evaluation/test_standard.py:41
↓ 1 callersMethod__init__
(self, labels, alpha: float, num_labeled: int, n_runs=1)
methods/pt_map/pt_map_loss.py:84
↓ 1 callersMethod__init__
(self, alpha=None, verbose: bool = False, progressBar: bool = False)
methods/pt_map/test_standard.py:128
↓ 1 callersMethod__init__
(self, image_size, normalize_param=dict(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]
methods/pt_map/data/datamgr.py:10
↓ 1 callersMethod__init__
(self, alpha=None)
methods/pt_map/evaluation/test_standard.py:124
↓ 1 callersMethod__init__
(self, block=BasicBlock, keep_prob=1.0, avg_pool=True, dropout=0.1, dropblock_size=5)
models/resnet12.py:74
↓ 1 callersMethod_compute_block_mask
(self, mask)
models/dropblock.py:30
↓ 1 callersFunction_load_pickle
(file)
methods/pt_map/FSLTask.py:24
↓ 1 callersMethod_make_layer
(self, block, in_planes, out_planes, nb_layers, stride, dropRate)
models/wrn_mixup_model.py:77
↓ 1 callersMethod_pairwise_dist
(a, b)
methods/pt_map/test_standard.py:94
↓ 1 callersMethodadjust_labeled
Adjust BPA scores using additional labels (e.g. support set in few shot classification) We do so by filling the final values of label
bpa/balanced_pairwise_affinities.py:50
↓ 1 callersFunctionbatched_log_sinkhorn
Batched version of log-space-sinkhorn.
bpa/ot.py:37
↓ 1 callersFunctioncenterDatas
Center labeled and un-labeled data separately.
methods/pt_map/pt_map_loss.py:11
↓ 1 callersFunctioncenterDatas
(datas)
methods/pt_map/test_standard.py:28
↓ 1 callersMethodclone
(self)
methods/pt_map/evaluation/test_standard.py:59
↓ 1 callersMethodcompute_cost_matrix
Compute the cost matrix under euclidean or cosine distances
bpa/balanced_pairwise_affinities.py:66
↓ 1 callersMethodcompute_optimal_transport
(self, M: torch.Tensor, r: torch.Tensor, c: torch.Tensor, epsilon: float = 1e-6)
methods/pt_map/pt_map_loss.py:41
↓ 1 callersMethodcompute_optimal_transport
(self, M, r, c, epsilon=1e-6)
methods/pt_map/test_standard.py:74
↓ 1 callersMethodcompute_optimal_transport
(self, M, r, c, epsilon=1e-6)
methods/pt_map/evaluation/test_standard.py:75
↓ 1 callersFunctionconv1x1
1x1 convolution
models/res_mixup_model.py:16
↓ 1 callersFunctiondownload_file
Helper method handling downloading large files from `url` to `filename`. Returns a pointer to `filename`.
datasets/get_cifar_fs.py:22
↓ 1 callersMethodestimateFromMask
(self, mask)
methods/pt_map/test_standard.py:118
↓ 1 callersMethodestimateFromMask
(self, mask)
methods/pt_map/evaluation/test_standard.py:112
↓ 1 callersMethodestimate_from_mask
(self, X: torch.Tensor, mask: torch.Tensor)
methods/pt_map/pt_map_loss.py:74
↓ 1 callersFunctionextract_feature
(val_loader, model, checkpoint_dir, tag='last')
methods/pt_map/save_plk.py:36
↓ 1 callersFunctionextract_feature
(val_loader, model, checkpoint_dir, tag='last')
methods/pt_map/evaluation/save_plk.py:47
↓ 1 callersMethodget_accuracy
(probas: torch.Tensor, labels: torch.Tensor)
methods/prototypical/proto_loss.py:21
↓ 1 callersMethodget_accuracy
(self, probas: torch.Tensor)
methods/pt_map/pt_map_loss.py:91
↓ 1 callersFunctionget_args
()
train.py:15
↓ 1 callersFunctionget_args
Description: Parses arguments at command line.
methods/pt_map/test_standard.py:177
↓ 1 callersMethodget_composed_transform
(self, aug=False)
methods/pt_map/data/datamgr.py:33
↓ 1 callersMethodget_data_loader
(self, data_file, aug)
methods/pt_map/data/datamgr.py:46
↓ 1 callersMethodget_data_loader
(self, data_file, aug)
methods/pt_map/data/datamgr.py:56
↓ 1 callersFunctionget_optimizer
Get optimizer.
utils.py:81
↓ 1 callersFunctionget_resume_file
(checkpoint_dir)
methods/pt_map/io_utils.py:43
↓ 1 callersFunctionget_scheduler
Get optimizer.
utils.py:95
↓ 1 callersMethodinitFromLabelledDatas
(self)
methods/pt_map/test_standard.py:66
↓ 1 callersMethodinitFromLabelledDatas
(self)
methods/pt_map/evaluation/test_standard.py:67
↓ 1 callersMethodinit_from_labelled
(self, X: torch.Tensor)
methods/pt_map/pt_map_loss.py:34
↓ 1 callersFunctionload_pickle
(file)
methods/pt_map/save_plk.py:31
↓ 1 callersFunctionload_pickle
(file)
methods/pt_map/evaluation/save_plk.py:43
↓ 1 callersMethodloop
(self, X: torch.Tensor, model: GaussianModel, n_epochs: int = 20)
methods/pt_map/pt_map_loss.py:104
↓ 1 callersMethodloop
(self, model, n_epochs=20)
methods/pt_map/test_standard.py:156
↓ 1 callersMethodloop
(self, model, n_epochs=20)
methods/pt_map/evaluation/test_standard.py:157
↓ 1 callersFunctionmain
()
train.py:115
↓ 1 callersFunctionmain
()
methods/pt_map/save_plk.py:64
↓ 1 callersMethodparse_csv
(self, txt_path)
datasets/cub.py:68
↓ 1 callersMethodparse_transform
(self, transform_type)
methods/pt_map/data/datamgr.py:17
↓ 1 callersMethodperformEpoch
(self, model, epochInfo=None)
methods/pt_map/test_standard.py:142
↓ 1 callersMethodperformEpoch
(self, model, epochInfo=None)
methods/pt_map/evaluation/test_standard.py:139
↓ 1 callersMethodperform_epoch
(self, model: GaussianModel, X: torch.Tensor)
methods/pt_map/pt_map_loss.py:98
↓ 1 callersFunctionsave_pickle
(file, data)
methods/pt_map/save_plk.py:26
↓ 1 callersFunctionsave_pickle
(file, data)
methods/pt_map/evaluation/save_plk.py:39
↓ 1 callersMethodscale
(self, X: torch.Tensor, mode: str)
methods/pt_map/pt_map_loss.py:124
↓ 1 callersFunctionsetRandomStates
(cfg)
methods/pt_map/FSLTask.py:95
↓ 1 callersFunctiontrain_one_epoch
(model, dataloader, optimizer, fewshot_method, criterion, labels, epoch, args)
train.py:202
↓ 1 callersMethodupdateFromEstimate
(self, estimate, alpha)
methods/pt_map/test_standard.py:69
↓ 1 callersMethodupdateFromEstimate
(self, estimate, alpha)
methods/pt_map/evaluation/test_standard.py:70
↓ 1 callersMethodupdate_from_estimate
(self, estimate, alpha)
methods/pt_map/pt_map_loss.py:37
FunctionGenerateRunSet
(start=None, end=None, cfg=None)
methods/pt_map/FSLTask.py:115
FunctionRes12
Constructs a ResNet-12 model.
models/resnet12.py:123
Method__call__
(self, img)
methods/pt_map/data/additional_transforms.py:19
Method__getattr__
(self, attr)
utils.py:352
Method__getitem__
(self, i)
methods/pt_map/data/dataset.py:23
Method__getitem__
(self, i)
methods/pt_map/data/dataset.py:57
Method__getitem__
(self, i)
methods/pt_map/data/dataset.py:71
Method__getitem__
(self, i)
datasets/cifar.py:53
Method__getitem__
(self, i)
datasets/cub.py:92
Method__getitem__
(self, i)
datasets/mini_imagenet.py:77
next →1–100 of 184, ranked by callers