Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ZhangYuanhan-AI/OmniBenchmark
/ types & classes
Types & classes
49 in github.com/ZhangYuanhan-AI/OmniBenchmark
⨍
Functions
269
◇
Types & classes
49
↓ 9 callers
Class
AverageMeter
Computes and stores the average and current value
ReCo/reco_imagenet.py:530
↓ 8 callers
Class
LayerNorm
Subclass torch's LayerNorm to handle fp16.
linear_probe/models/text_transformer.py:15
↓ 3 callers
Class
ModifiedResNet
A ResNet class that is similar to torchvision's but contains the following changes: - There are now 3 "stem" convolutions as opposed to 1, wi
linear_probe/models/modified_resnet.py:157
↓ 3 callers
Class
Mybeit
linear_probe/models/timm_beit.py:20
↓ 3 callers
Class
VisionTransformer
linear_probe/models/vit.py:260
↓ 2 callers
Class
AutoAugmentOp
ReCo/randaugment.py:284
↓ 2 callers
Class
Bottleneck
linear_probe/models/vit.py:68
↓ 2 callers
Class
Bottleneck
linear_probe/models/modified_resnet.py:74
↓ 2 callers
Class
MyViT
linear_probe/models/timm_vit_modified.py:54
↓ 2 callers
Class
MyViT
linear_probe/models/timm_vit.py:22
↓ 2 callers
Class
Myeffnet
linear_probe/models/timm_effnet.py:20
↓ 2 callers
Class
Myinceptionv4
linear_probe/models/timm_inceptionv4.py:20
↓ 2 callers
Class
Mymlp_mixer
linear_probe/models/timm_mlp_mixer.py:20
↓ 2 callers
Class
MyswimTransformer
linear_probe/models/timm_swinTransformer.py:10
↓ 2 callers
Class
ProgressMeter
ReCo/reco_imagenet.py:554
↓ 2 callers
Class
ReCoLoss
ReCo/losses.py:105
↓ 2 callers
Class
ResNet
linear_probe/models/resnet.py:96
↓ 2 callers
Class
Transformer
linear_probe/models/vit.py:249
↓ 1 callers
Class
AttentionPool2d
linear_probe/models/vit.py:114
↓ 1 callers
Class
AttentionPool2d
linear_probe/models/modified_resnet.py:120
↓ 1 callers
Class
CLIP_text
linear_probe/models/text_transformer.py:78
↓ 1 callers
Class
ImageReader_ceph
linear_probe/dataset.py:30
↓ 1 callers
Class
ModifiedResNet
A ResNet class that is similar to torchvision's but contains the following changes: - There are now 3 "stem" convolutions as opposed to 1, wi
linear_probe/models/vit.py:151
↓ 1 callers
Class
Myrn101
linear_probe/models/timm_rn101.py:20
↓ 1 callers
Class
QuickGELU
linear_probe/models/vit.py:220
↓ 1 callers
Class
QuickGELU
linear_probe/models/text_transformer.py:26
↓ 1 callers
Class
RandAugment
ReCo/randaugment.py:393
↓ 1 callers
Class
ResNet
ReCo/models/resnet.py:143
↓ 1 callers
Class
ResidualAttentionBlock
linear_probe/models/vit.py:225
↓ 1 callers
Class
ResidualAttentionBlock
linear_probe/models/text_transformer.py:31
↓ 1 callers
Class
Transformer
linear_probe/models/text_transformer.py:55
Class
AverageMeter
Computes and stores the average and current value
linear_probe/utils.py:60
Class
BasicBlock
linear_probe/models/resnet.py:25
Class
BasicBlock
ReCo/models/resnet.py:36
Class
Bottleneck
linear_probe/models/resnet.py:57
Class
Bottleneck
ReCo/models/resnet.py:85
Class
CLIP
linear_probe/models/vit.py:299
Class
GaussianBlur
blur a single image on CPU
ReCo/randaugment.py:454
Class
GaussianBlur
Gaussian blur augmentation in SimCLR https://arxiv.org/abs/2002.05709
ReCo/moco/loader.py:29
Class
MoCo
Build a MoCo model with: a query encoder, a key encoder, and a queue https://arxiv.org/abs/1911.05722
ReCo/moco/builder.py:22
Class
NormedLinear_Classifier
ReCo/moco/builder.py:7
Class
PaCoLoss
ReCo/losses.py:7
Class
SimpleTokenizer
linear_probe/models/simple_tokenizer.py:62
Class
TwoCropsTransform
Take two random crops of one image as the query and key.
ReCo/moco/loader.py:6
Class
TwoCropsTransformV2
Take two random crops of one image as the query and key.
ReCo/moco/loader.py:17
Class
general_dataset
linear_probe/dataset.py:303
Class
memcached_DataFolder
A generic data loader where the samples are arranged in this way: :: root/class_x/xxx.ext root/class_x/xxy.ext root/class_x/x
linear_probe/dataset.py:102
Class
memcached_ImageFolder
A generic data loader where the images are arranged in this way: :: root/dog/xxx.png root/dog/xxy.png root/dog/xxz.png
linear_probe/dataset.py:218
Class
memcached_dataset
linear_probe/dataset.py:249