MCPcopy Create free account

hub / github.com/LouieYang/MCL / functions

Functions259 in github.com/LouieYang/MCL

↓ 16 callersMethodstep
(self, epoch=None, metrics=None)
engines/utils.py:98
↓ 11 callersFunction_l2norm
(x, dim=1, keepdim=True)
modules/utils.py:3
↓ 11 callersMethodupdate
(self, val, n=1)
engines/utils.py:41
↓ 9 callersFunctionset_seed
(seed, print_=True)
engines/utils.py:7
↓ 8 callersFunctionget_world_size
()
engines/distributed_utils.py:4
↓ 6 callersFunctionmake_fsl
(cfg)
modules/fsl_query.py:109
↓ 6 callersFunctionmean_confidence_interval
(data, confidence=0.95)
engines/utils.py:23
↓ 6 callersMethodrun
(self)
engines/trainer.py:178
↓ 5 callersFunctioncfg_to_dataset
(cfg)
experiments/utils.py:1
↓ 5 callersFunctionmake_distributed_dataloader
(cfg, phase, batch_size, distributed_info, epoch=0, pretrain=False)
dataloader/__init__.py:63
↓ 5 callersMethodmerge_from_file
Load a yaml config file and merge it into the global config.
configs/collections.py:55
↓ 5 callersFunctionreduce_loss_dict
Reduce the loss dictionary from all processes so that process with rank 0 has the averaged results. Returns a dict with the same fields as
engines/distributed_utils.py:35
↓ 4 callersMethod_make_layer
(self, block, planes, stride=1, drop_rate=0.0, drop_block=False, block_size=1)
modules/encoder/resnet.py:146
↓ 4 callersFunctionis_main_process
()
engines/distributed_utils.py:18
↓ 3 callersMethod_wide_layer
(self, block, planes, num_blocks, dropout_rate, stride)
modules/encoder/wrn28.py:64
↓ 3 callersFunctionconv3x3
3x3 convolution with padding
modules/encoder/resnet.py:67
↓ 3 callersFunctionl2distance
Input: x [.., c, M_x] y [.., c, M_y] Return: ret [.., M_x, M_y]
modules/utils.py:6
↓ 3 callersFunctionmake_dataloader
(cfg, phase, batch_size=1, save_summary_dir=None, load_summary_dir=None)
dataloader/__init__.py:37
↓ 3 callersMethodmerge_from_list
Merge config keys, values in a list (e.g., from command line) into the global config. For example, `cfg_list = ['TEST.NMS', 0.5]`.
configs/collections.py:62
↓ 2 callersMethod__init__
(self, block=BasicBlock, drop_rate=0.0, dropblock_size=3)
modules/encoder/resnet.py:129
↓ 2 callersFunction_check_and_coerce_cfg_value_type
Checks that `value_a`, which is intended to replace `value_b` is of the right type. The type is correct if it matches exactly or is one of a few
configs/collections.py:150
↓ 2 callersFunction_decide_dataset
(cfg, phase, save_summary_dir=None, load_summary_dir=None)
dataloader/__init__.py:16
↓ 2 callersFunction_decode_cfg_value
Decodes a raw config value (e.g., from a yaml config files or command line argument) into a Python object.
configs/collections.py:118
↓ 2 callersMethod_get_griditems
(self, img)
dataloader/grid_datasets.py:75
↓ 2 callersMethod_inner_state_similarity
(self, states, mask_diagonal=None)
modules/query/ls.py:36
↓ 2 callersMethod_pyramid_encoding
(self, x)
modules/fsl_query.py:28
↓ 2 callersFunction_register_generic
(module_dict, module_name, module)
modules/utils.py:79
↓ 2 callersFunctionbatched_index_select
(input_, dim, index)
modules/utils.py:32
↓ 2 callersMethodbatched_kronecker
(self, matrix1, matrix2)
modules/query/metaoptnet.py:63
↓ 2 callersMethodcomputeGramMatrix
Constructs a linear kernel matrix between A and B. We assume that each row in A and B represents a d-dimensional feature vector.
modules/query/metaoptnet.py:46
↓ 2 callersMethodemd_forward_1shot
(self, support_xf, query_xf, n_way)
modules/query/deepemd.py:117
↓ 2 callersMethodget_grid_location
:param size: size of the height/width :param ratio: generate grid size/ even divided grid size :param num_grid: number of gri
dataloader/grid_datasets.py:32
↓ 2 callersMethodget_recon_dist
(self,query,support,alpha,beta,Woodbury=True)
modules/pretrainer/frn.py:24
↓ 2 callersMethodget_weight_vector
(self, A, B)
modules/query/deepemd.py:54
↓ 2 callersFunctionmake_encoder
(cfg)
modules/encoder/__init__.py:8
↓ 2 callersFunctionmake_predataloader
(cfg, phase, batch_size=1)
dataloader/__init__.py:44
↓ 2 callersFunctionmake_pretrain_model
(cfg)
modules/pretrainer/__init__.py:6
↓ 2 callersMethodnormalize_feature
(self, x)
modules/query/deepemd.py:73
↓ 2 callersMethodsave_model
(self, prefix, accuracy, h, epoch, final_epoch=False)
engines/trainer.py:138
↓ 2 callersMethodsave_summary_datalist
(self, save_dir)
dataloader/base_datasets.py:106
↓ 2 callersFunctionsynchronize
Helper function to synchronize (barrier) among all processes when using distributed training
engines/distributed_utils.py:21
↓ 2 callersMethodtrain
(self, dataloader, epoch)
engines/distributed_trainer.py:162
↓ 2 callersMethodtrain
(self, dataloader, epoch)
engines/pretrainer.py:104
↓ 2 callersMethodtrain
(self, dataloader, epoch)
engines/distributed_pretrainer.py:118
↓ 2 callersMethodtrain
(self, dataloader, epoch)
engines/trainer.py:155
↓ 1 callersMethod_1MNN_mask
(self, S)
modules/query/mn4.py:28
↓ 1 callersMethod__init__
(self, in_planes, planes, dropout_rate, stride=1)
modules/encoder/wrn28.py:25
↓ 1 callersMethod__init__
(self, in_channels, cfg)
modules/query/relationnet.py:31
↓ 1 callersMethod__init__
(self, indim, outdim)
modules/query/linearplus.py:11
↓ 1 callersMethod__init__
(self, in_features, out_features, scale_factor=10.0, margin=0.40)
modules/query/negative_softmax.py:17
↓ 1 callersMethod__init__
(self, in_channels, cfg)
modules/query/relationmel.py:33
↓ 1 callersMethod__init__
(self, in_channels, cfg)
modules/query/negative_cosine.py:42
↓ 1 callersMethod__init__
(self)
engines/utils.py:32
↓ 1 callersMethod_compute_block_mask
(self, mask)
modules/encoder/resnet.py:30
↓ 1 callersMethod_equilibrium_state
(self, T_sq, T_qs)
modules/query/mel.py:21
↓ 1 callersFunction_merge_a_into_b
Merge config dictionary a into config dictionary b, clobbering the options in b whenever they are also specified in a.
configs/collections.py:85
↓ 1 callersMethod_parse_encoding_params
(self)
modules/fsl_query.py:20
↓ 1 callersMethod_parse_encoding_params
(self)
dataloader/grid_datasets.py:24
↓ 1 callersMethod_prepare_summary_snapshots
(self, prefix, cfg)
engines/distributed_trainer.py:105
↓ 1 callersMethod_prepare_summary_snapshots
(self, prefix, cfg)
engines/pretrainer.py:82
↓ 1 callersMethod_prepare_summary_snapshots
(self, prefix, cfg)
engines/distributed_pretrainer.py:93
↓ 1 callersMethod_prepare_summary_snapshots
(self, prefix, cfg)
engines/trainer.py:93
↓ 1 callersMethod_scores
(self, support_xf, support_y, query_xf, query_y)
modules/query/relationnet.py:48
↓ 1 callersMethod_scores
(self, support_xf, support_y, query_xf, query_y)
modules/query/protonet.py:29
↓ 1 callersMethod_scores
(self, support_xf, support_y, query_xf, query_y)
modules/query/matchingnet.py:29
↓ 1 callersFunctionattach_dim
(v, n_dim_to_prepend=0, n_dim_to_append=0)
modules/utils.py:73
↓ 1 callersMethodaveraging_based_similarities
(self, support_xf, support_y, query_xf, query_y)
modules/query/ls.py:25
↓ 1 callersMethodbinv
Computes an inverse of each matrix in the batch. Pytorch 0.4.1 does not support batched matrix inverse. Hence, we are solving
modules/query/r2d2.py:54
↓ 1 callersMethodbipartite_katz_forward
(self, support_xf, support_y, query_xf, query_y, similarity_f)
modules/query/mel.py:55
↓ 1 callersMethodcomputeGramMatrix
Constructs a linear kernel matrix between A and B. We assume that each row in A and B represents a d-dimensional feature vector.
modules/query/r2d2.py:37
↓ 1 callersFunctionconv3x3
(in_planes, out_planes, stride=1)
modules/encoder/wrn28.py:12
↓ 1 callersMethodemd_inference_opencv
(self, cost_matrix, weight1, weight2)
modules/query/deepemd.py:41
↓ 1 callersMethodfix_bn
(self)
engines/distributed_trainer.py:116
↓ 1 callersMethodfix_bn
(self)
engines/trainer.py:101
↓ 1 callersMethodforward_FCN
(self, support_x, support_y, query_x, query_y)
modules/fsl_query.py:68
↓ 1 callersMethodforward_Grid
(self, support_x, support_y, query_x, query_y)
modules/fsl_query.py:41
↓ 1 callersMethodforward_PyramidFCN
(self, support_x, support_y, query_x, query_y)
modules/fsl_query.py:54
↓ 1 callersMethodforward_per_batch
query B x D x H x W support Nc x Nk x Ds x Hs x Ws (#CLASSES x #SHOT x #DIMENSIONS)
modules/query/ctx.py:33
↓ 1 callersMethodforward_test
(self, support_x, support_y, query_x, query_y)
modules/pretrainer/base.py:24
↓ 1 callersMethodforward_train
(self, x, y)
modules/pretrainer/base.py:21
↓ 1 callersMethodget_emd_distance
(self, similarity_map, weight_1, weight_2)
modules/query/deepemd.py:77
↓ 1 callersMethodget_neg_l2_dist
(self, support_xf, query_xf)
modules/query/frn.py:69
↓ 1 callersMethodget_neg_l2_dist
(self, support_xf, query_xf)
modules/pretrainer/frn.py:59
↓ 1 callersMethodget_pyramid
(self, img, num_grid)
dataloader/grid_datasets.py:52
↓ 1 callersFunctionget_rank
()
engines/distributed_utils.py:11
↓ 1 callersMethodget_recon_dist
(self,query,support,alpha,beta,Woodbury=True)
modules/query/frn.py:33
↓ 1 callersMethodget_sfc
(self, support_xf, support_y, n_way, k_shot)
modules/query/deepemd.py:92
↓ 1 callersMethodhorizontal_flip_aug
(self, support_x, support_y, k_shot)
modules/fsl_query.py:80
↓ 1 callersFunctionmain
()
experiments/run_trainer.py:14
↓ 1 callersFunctionmain
()
experiments/run_distributed_trainer.py:15
↓ 1 callersFunctionmain
()
experiments/run_distributed_pretrainer.py:15
↓ 1 callersFunctionmain
()
experiments/run_pretrainer.py:13
↓ 1 callersFunctionmain
()
experiments/run_evaluator.py:14
↓ 1 callersFunctionmake_query
(in_channels, cfg)
modules/query/__init__.py:32
↓ 1 callersMethodpooling
(self, support_xf, support_y, query_xf, query_y, n_way, k_shot)
modules/query/r2d2.py:29
↓ 1 callersMethodpooling
(self, support_xf, support_y, query_xf, query_y, n_way, k_shot)
modules/query/linearplus.py:55
↓ 1 callersMethodpooling
(self, support_xf, support_y, query_xf, query_y, n_way, k_shot)
modules/query/negative_softmax.py:61
↓ 1 callersMethodpooling
(self, support_xf, support_y, query_xf, query_y, n_way, k_shot)
modules/query/metaoptnet.py:38
↓ 1 callersMethodpooling
(self, support_xf, support_y, query_xf, query_y, n_way, k_shot)
modules/query/negative_cosine.py:63
↓ 1 callersMethodpooling
(self, support_xf, support_y, query_xf, query_y, n_way, k_shot)
modules/query/dsn.py:28
next →1–100 of 259, ranked by callers