Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/TongZhangTHU/sgr
/ types & classes
Types & classes
54 in github.com/TongZhangTHU/sgr
⨍
Functions
316
◇
Types & classes
54
↓ 34 callers
Class
DenseBlock
helpers/network_utils.py:257
↓ 14 callers
Class
Conv3DBlock
helpers/network_utils.py:128
↓ 10 callers
Class
Conv2DBlock
helpers/network_utils.py:67
↓ 9 callers
Class
Conv2DFiLMBlock
helpers/network_utils.py:110
↓ 5 callers
Class
LayerNorm
Subclass torch's LayerNorm to handle fp16.
helpers/clip/core/clip.py:210
↓ 5 callers
Class
PreNorm
agents/peract_bc/perceiver_lang_io.py:52
↓ 3 callers
Class
Attention
agents/peract_bc/perceiver_lang_io.py:89
↓ 3 callers
Class
Lamb
r"""Implements Lamb algorithm. It has been proposed in `Large Batch Optimization for Deep Learning: Training BERT in 76 minutes`_. Arguments:
helpers/optim/lamb.py:23
↓ 3 callers
Class
PreprocessAgent
helpers/preprocess_agent.py:9
↓ 3 callers
Class
SpatialSoftmax3D
helpers/network_utils.py:770
↓ 2 callers
Class
Bottleneck
helpers/clip/core/clip.py:45
↓ 2 callers
Class
Conv3DInceptionBlock
helpers/network_utils.py:705
↓ 2 callers
Class
FeedForward
agents/peract_bc/perceiver_lang_io.py:76
↓ 2 callers
Class
PreNorm
helpers/network_utils.py:483
↓ 2 callers
Class
Transformer
helpers/clip/core/clip.py:248
↓ 1 callers
Class
Attention
helpers/network_utils.py:506
↓ 1 callers
Class
AttentionPool2d
helpers/clip/core/clip.py:91
↓ 1 callers
Class
CLIP
helpers/clip/core/clip.py:311
↓ 1 callers
Class
Conv3DUpsampleBlock
helpers/network_utils.py:237
↓ 1 callers
Class
ConvTranspose3DBlock
helpers/network_utils.py:173
↓ 1 callers
Class
FeedForward
helpers/network_utils.py:492
↓ 1 callers
Class
FiLMBlock
helpers/network_utils.py:54
↓ 1 callers
Class
GEGLU
agents/peract_bc/perceiver_lang_io.py:70
↓ 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
helpers/clip/core/clip.py:128
↓ 1 callers
Class
PerceiverVoxelLangEncoder
agents/peract_bc/perceiver_lang_io.py:132
↓ 1 callers
Class
QAttentionPerActBCAgent
agents/peract_bc/qattention_peract_bc_agent.py:110
↓ 1 callers
Class
QAttentionStackAgent
agents/peract_bc/qattention_stack_agent.py:15
↓ 1 callers
Class
QFunction
agents/peract_bc/qattention_peract_bc_agent.py:30
↓ 1 callers
Class
QFunction
agents/semantic_bc_lang/semantic_bc_lang_agent.py:18
↓ 1 callers
Class
QuickGELU
helpers/clip/core/clip.py:219
↓ 1 callers
Class
ResidualAttentionBlock
helpers/clip/core/clip.py:224
↓ 1 callers
Class
Resnet_SGR_Modified
helpers/resnet.py:12
↓ 1 callers
Class
SGRAgent
agents/sgr/sgr_agent.py:109
↓ 1 callers
Class
SGRNetWrapper
agents/sgr/sgr_agent.py:21
↓ 1 callers
Class
SGRNetwork
agents/sgr/sgr_network.py:14
↓ 1 callers
Class
SemanticBCLangAgent
agents/semantic_bc_lang/semantic_bc_lang_agent.py:94
↓ 1 callers
Class
SemanticCNNLangAndFcsNet
agents/semantic_bc_lang/networks.py:10
↓ 1 callers
Class
SiameseNet
helpers/network_utils.py:292
↓ 1 callers
Class
StackAgent
agents/sgr/stack_agent.py:10
↓ 1 callers
Class
StackAgent
agents/semantic_bc_lang/stack_agent.py:9
↓ 1 callers
Class
Transformer
helpers/network_utils.py:539
↓ 1 callers
Class
VisualTransformer
helpers/clip/core/clip.py:259
↓ 1 callers
Class
VoxelGrid
voxel/voxel_grid.py:12
Class
CNNAndFcsNet
helpers/network_utils.py:333
Class
CNNLangAndFcsNet
helpers/network_utils.py:393
Class
Conv2DUpsampleBlock
helpers/network_utils.py:217
Class
Conv3DInceptionBlockUpsampleBlock
helpers/network_utils.py:679
Class
ConvTransposeUp3DBlock
helpers/network_utils.py:745
Class
CustomMultiTaskRLBenchEnv
helpers/custom_rlbench_env.py:206
Class
CustomMultiVariationRLBenchEnv
helpers/custom_rlbench_env.py:397
Class
CustomRLBenchEnv
helpers/custom_rlbench_env.py:20
Class
SimpleTokenizer
helpers/clip/core/simple_tokenizer.py:62
Class
ViT
helpers/network_utils.py:555
Class
ViTLangAndFcsNet
helpers/network_utils.py:595