Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Monoxide-Chen/uncertainty_retrieval
/ types & classes
Types & classes
75 in github.com/Monoxide-Chen/uncertainty_retrieval
⨍
Functions
315
◇
Types & classes
75
↓ 4 callers
Class
SimpleVocabulary
language/vocabulary.py:31
↓ 3 callers
Class
AttentionModule
models/attention_modules/self_attention.py:7
↓ 3 callers
Class
AverageMeterSet
utils/metrics.py:1
↓ 3 callers
Class
SimpleLSTMEncoder
models/text_encoders/lstm.py:9
↓ 2 callers
Class
AverageMeter
Computes and stores the average and current value
utils/metrics.py:34
↓ 2 callers
Class
EqualLinear
models/utils.py:49
↓ 2 callers
Class
GlobalCrossAttentionMap
models/attention_modules/self_attention.py:71
↓ 2 callers
Class
LoggingService
trainers/abc.py:19
↓ 2 callers
Class
NormalizationLSTMEncoder
models/text_encoders/lstm.py:38
↓ 2 callers
Class
SelfAttentionMap
models/attention_modules/self_attention.py:45
↓ 2 callers
Class
SimplerLSTMEncoder
models/text_encoders/lstm.py:52
↓ 2 callers
Class
ToIds
transforms/text_transforms.py:9
↓ 2 callers
Class
ToLongTensor
transforms/text_transforms.py:17
↓ 2 callers
Class
WandbSimplePrinter
loggers/wandb_loggers.py:6
↓ 1 callers
Class
AleatoricLoss
losses/uncertainty_loss.py:8
↓ 1 callers
Class
BasicTokenizer
language/tokenizers.py:11
↓ 1 callers
Class
BatchBasedAleatoricLoss
losses/uncertainty_loss.py:36
↓ 1 callers
Class
BatchBasedClassificationLoss
losses/batch_based_classification_loss.py:7
↓ 1 callers
Class
BertFc
models/text_encoders/roberta.py:31
↓ 1 callers
Class
BertPaddingCollateFunction
data/collate_fns.py:46
↓ 1 callers
Class
BertPaddingCollateFunctionTest
data/collate_fns.py:69
↓ 1 callers
Class
BestModelTracker
loggers/file_loggers.py:36
↓ 1 callers
Class
CAWRwithWarmup
Args: optimizer (Optimizer): Wrapped optimizer. T_0 (int): Number of iterations for the first restart. Default: 1. T_mult (int
optimizers.py:7
↓ 1 callers
Class
DisentangledTransformer
models/compositors/transformers.py:7
↓ 1 callers
Class
EqualLR
models/utils.py:17
↓ 1 callers
Class
Fashion200kDataset
data/fashion200k.py:41
↓ 1 callers
Class
Fashion200kTestDataset
data/fashion200k.py:202
↓ 1 callers
Class
Fashion200kTestQueryDataset
data/fashion200k.py:279
↓ 1 callers
Class
FashionIQDataset
Fashion200K dataset. Image pairs in {root_path}/image_pairs/{split}_pairs.pkl
data/fashionIQ.py:77
↓ 1 callers
Class
FashionIQTestDataset
FashionIQ Test (Samples) dataset. indexing returns target samples and their unique ID
data/fashionIQ.py:125
↓ 1 callers
Class
FashionIQTestQueryDataset
FashionIQ Test (Query) dataset. indexing returns ref samples, modifier, target attribute (caption, text) and modifier length
data/fashionIQ.py:165
↓ 1 callers
Class
GlobalStyleTransformer2
models/compositors/global_style_models.py:7
↓ 1 callers
Class
GradientControlDataParallel
utils/mixins.py:37
↓ 1 callers
Class
NormalAugmenter
add Gaussian noisy to feature
models/augmenter/normal_augmenter.py:4
↓ 1 callers
Class
PaddingCollateFunction
data/collate_fns.py:7
↓ 1 callers
Class
PaddingCollateFunctionTest
data/collate_fns.py:22
↓ 1 callers
Class
ResNet18Layer4Lower
models/image_encoders/resnet.py:11
↓ 1 callers
Class
ResNet18Layer4Upper
models/image_encoders/resnet.py:33
↓ 1 callers
Class
ResNet50Layer4Lower
models/image_encoders/resnet.py:62
↓ 1 callers
Class
ResNet50Layer4Upper
models/image_encoders/resnet.py:88
↓ 1 callers
Class
RobertaEncoder
models/text_encoders/roberta.py:7
↓ 1 callers
Class
ShoesDataset
shoes dataset root_path = datasets/shoes/attributedata clothing=None (Not used in shoes dataset)
data/shoes.py:44
↓ 1 callers
Class
ShoesTestDataset
Shoes Test (Samples) dataset. indexing returns target samples and their unique ID
data/shoes.py:90
↓ 1 callers
Class
ShoesTestQueryDataset
Shoes Test (Query) dataset. indexing returns ref samples, modifier, target attribute (caption, text) and modifier length
data/shoes.py:147
↓ 1 callers
Class
ValidationMetricsCalculator
evaluators/metric_calculators.py:10
↓ 1 callers
Class
WandbSummaryPrinter
loggers/wandb_loggers.py:18
↓ 1 callers
Class
_enable_get_lr_call
optimizers.py:77
Class
AbstractBaseCompositor
trainers/abc.py:80
Class
AbstractBaseDataset
Base class for a dataset.
data/abc.py:7
Class
AbstractBaseEvaluator
evaluators/abc.py:12
Class
AbstractBaseFashion200kDataset
data/fashion200k.py:27
Class
AbstractBaseFashionIQDataset
data/fashionIQ.py:58
Class
AbstractBaseImageLowerEncoder
trainers/abc.py:43
Class
AbstractBaseImageUpperEncoder
trainers/abc.py:53
Class
AbstractBaseLogger
trainers/abc.py:9
Class
AbstractBaseMetricLoss
trainers/abc.py:32
Class
AbstractBaseShoesDataset
data/shoes.py:29
Class
AbstractBaseTextEncoder
trainers/abc.py:64
Class
AbstractBaseTokenizer
language/vocabulary.py:10
Class
AbstractBaseTrainer
trainers/abc.py:102
Class
AbstractBaseVocabulary
language/abc.py:12
Class
AbstractGlobalStyleTransformer
trainers/abc.py:91
Class
AbstractGradientControl
utils/mixins.py:7
Class
GAPResNet18Layer4Upper
models/image_encoders/resnet.py:48
Class
GradientControlMixin
utils/mixins.py:17
Class
NltkTokenizer
language/tokenizers.py:20
Class
RecentModelTracker
loggers/file_loggers.py:23
Class
SimpleEvaluator
evaluators/tirg_evaluator.py:4
Class
TIRGTrainer
trainers/tirg_trainer.py:7
Class
TestBasicTokenizer
language/test_tokenizers.py:7
Class
TestLSTMModel
models/text_encoders/test_lstm.py:8
Class
TestNltkTokenizer
language/test_tokenizers.py:18
Class
TestSelfAttentionMap
models/attention_modules/test_self_attention.py:8
Class
TestSimpleVocabulary
language/test_vocabulary.py:7
Class
TestTextModelUtils
models/text_encoders/test_utils.py:8