Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/JialianW/GRiT
/ types & classes
Types & classes
41 in github.com/JialianW/GRiT
⨍
Functions
187
◇
Types & classes
41
↓ 11 callers
Class
ObjDescription
grit/data/custom_dataset_mapper.py:129
↓ 3 callers
Class
ViT
This module implements Vision Transformer (ViT) backbone in :paper:`vitdet`. "Exploring Plain Vision Transformer Backbones for Object Detecti
grit/modeling/backbone/vit.py:237
↓ 3 callers
Class
ViT_FPN
grit/modeling/backbone/vit.py:366
↓ 2 callers
Class
BertIntermediate
grit/modeling/text/modeling_bert.py:177
↓ 2 callers
Class
BertOutput
grit/modeling/text/modeling_bert.py:190
↓ 2 callers
Class
CustomDatasetMapper
grit/data/custom_dataset_mapper.py:20
↓ 2 callers
Class
GRiTCOCOEvaluator
grit/evaluation/eval.py:12
↓ 2 callers
Class
GRiTVGEvaluator
grit/evaluation/eval.py:86
↓ 2 callers
Class
Mlp
grit/modeling/text/modeling_bert.py:209
↓ 1 callers
Class
Attention
Multi-head Attention block with relative position embeddings.
grit/modeling/backbone/vit.py:32
↓ 1 callers
Class
AutoRegressiveBeamSearch
grit/modeling/text/text_decoder.py:356
↓ 1 callers
Class
BertAttention
grit/modeling/text/modeling_bert.py:155
↓ 1 callers
Class
BertConfig
r""" :class:`~pytorch_transformers.BertConfig` is the configuration class to store the configuration of a `BertModel`. Argum
grit/modeling/text/modeling_bert.py:458
↓ 1 callers
Class
BertEncoder
grit/modeling/text/modeling_bert.py:263
↓ 1 callers
Class
BertEncoderAsDecoder
grit/modeling/text/text_decoder.py:74
↓ 1 callers
Class
BertLayer
grit/modeling/text/modeling_bert.py:227
↓ 1 callers
Class
BertSelfAttention
grit/modeling/text/modeling_bert.py:70
↓ 1 callers
Class
BertSelfOutput
grit/modeling/text/modeling_bert.py:136
↓ 1 callers
Class
Block
Transformer blocks with support of window attention and residual propagation blocks
grit/modeling/backbone/vit.py:152
↓ 1 callers
Class
EfficientDetResizeCrop
Scale the shorter edge to the given size, with a limit of `max_size` on the longer edge. If `max_size` is reached, then downscale so that the
grit/data/transforms/custom_augmentation_impl.py:17
↓ 1 callers
Class
EfficientDetResizeCropTransform
grit/data/transforms/custom_transform.py:29
↓ 1 callers
Class
GRiTFastRCNNOutputLayers
grit/modeling/roi_heads/grit_fast_rcnn.py:17
↓ 1 callers
Class
GRiTTextDecoder
grit/modeling/text/text_decoder.py:542
↓ 1 callers
Class
LoadTextTokens
grit/modeling/text/load_text_token.py:4
↓ 1 callers
Class
MultiDatasetSampler
grit/data/custom_dataset_dataloader.py:157
↓ 1 callers
Class
MultiDatasets
grit/data/custom_dataset_dataloader.py:230
↓ 1 callers
Class
PatchEmbed
Image to Patch Embedding.
grit/modeling/backbone/utils.py:160
↓ 1 callers
Class
QK2Attention
grit/modeling/text/modeling_bert.py:62
↓ 1 callers
Class
ResBottleneckBlock
The standard bottleneck residual block without the last activation layer. It contains 3 conv layers with kernels 1x1, 3x3, 1x1.
grit/modeling/backbone/vit.py:91
↓ 1 callers
Class
SingleDataset
grit/data/custom_dataset_dataloader.py:212
↓ 1 callers
Class
SmoothLabelCrossEntropyLoss
grit/modeling/text/text_decoder.py:644
↓ 1 callers
Class
TransformerDecoderTextualHead
grit/modeling/text/text_decoder.py:229
↓ 1 callers
Class
VisualizationDemo
grit/predictor.py:51
↓ 1 callers
Class
Visualizer_GRiT
grit/predictor.py:9
↓ 1 callers
Class
WordAndPositionalEmbedding
grit/modeling/text/text_decoder.py:24
Class
FullModelGradientClippingOptimizer
grit/custom_solver.py:44
Class
GRiT
grit/modeling/meta_arch/grit.py:10
Class
GRiTROIHeadsAndTextDecoder
grit/modeling/roi_heads/grit_roi_heads.py:27
Class
PreNormTransformerDecoderLayer
grit/modeling/text/text_decoder.py:201
Class
PretrainedConfig
Base class for all configuration classes. Handle a few common parameters and methods for loading/downloading/saving configurations.
grit/modeling/text/modeling_bert.py:302
Class
TextualHead
grit/modeling/text/text_decoder.py:11