MCPcopy Create free account

hub / github.com/TongZhangTHU/sgr / types & classes

Types & classes54 in github.com/TongZhangTHU/sgr

↓ 34 callersClassDenseBlock
helpers/network_utils.py:257
↓ 14 callersClassConv3DBlock
helpers/network_utils.py:128
↓ 10 callersClassConv2DBlock
helpers/network_utils.py:67
↓ 9 callersClassConv2DFiLMBlock
helpers/network_utils.py:110
↓ 5 callersClassLayerNorm
Subclass torch's LayerNorm to handle fp16.
helpers/clip/core/clip.py:210
↓ 5 callersClassPreNorm
agents/peract_bc/perceiver_lang_io.py:52
↓ 3 callersClassAttention
agents/peract_bc/perceiver_lang_io.py:89
↓ 3 callersClassLamb
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 callersClassPreprocessAgent
helpers/preprocess_agent.py:9
↓ 3 callersClassSpatialSoftmax3D
helpers/network_utils.py:770
↓ 2 callersClassBottleneck
helpers/clip/core/clip.py:45
↓ 2 callersClassConv3DInceptionBlock
helpers/network_utils.py:705
↓ 2 callersClassFeedForward
agents/peract_bc/perceiver_lang_io.py:76
↓ 2 callersClassPreNorm
helpers/network_utils.py:483
↓ 2 callersClassTransformer
helpers/clip/core/clip.py:248
↓ 1 callersClassAttention
helpers/network_utils.py:506
↓ 1 callersClassAttentionPool2d
helpers/clip/core/clip.py:91
↓ 1 callersClassCLIP
helpers/clip/core/clip.py:311
↓ 1 callersClassConv3DUpsampleBlock
helpers/network_utils.py:237
↓ 1 callersClassConvTranspose3DBlock
helpers/network_utils.py:173
↓ 1 callersClassFeedForward
helpers/network_utils.py:492
↓ 1 callersClassFiLMBlock
helpers/network_utils.py:54
↓ 1 callersClassGEGLU
agents/peract_bc/perceiver_lang_io.py:70
↓ 1 callersClassModifiedResNet
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 callersClassPerceiverVoxelLangEncoder
agents/peract_bc/perceiver_lang_io.py:132
↓ 1 callersClassQAttentionPerActBCAgent
agents/peract_bc/qattention_peract_bc_agent.py:110
↓ 1 callersClassQAttentionStackAgent
agents/peract_bc/qattention_stack_agent.py:15
↓ 1 callersClassQFunction
agents/peract_bc/qattention_peract_bc_agent.py:30
↓ 1 callersClassQFunction
agents/semantic_bc_lang/semantic_bc_lang_agent.py:18
↓ 1 callersClassQuickGELU
helpers/clip/core/clip.py:219
↓ 1 callersClassResidualAttentionBlock
helpers/clip/core/clip.py:224
↓ 1 callersClassResnet_SGR_Modified
helpers/resnet.py:12
↓ 1 callersClassSGRAgent
agents/sgr/sgr_agent.py:109
↓ 1 callersClassSGRNetWrapper
agents/sgr/sgr_agent.py:21
↓ 1 callersClassSGRNetwork
agents/sgr/sgr_network.py:14
↓ 1 callersClassSemanticBCLangAgent
agents/semantic_bc_lang/semantic_bc_lang_agent.py:94
↓ 1 callersClassSemanticCNNLangAndFcsNet
agents/semantic_bc_lang/networks.py:10
↓ 1 callersClassSiameseNet
helpers/network_utils.py:292
↓ 1 callersClassStackAgent
agents/sgr/stack_agent.py:10
↓ 1 callersClassStackAgent
agents/semantic_bc_lang/stack_agent.py:9
↓ 1 callersClassTransformer
helpers/network_utils.py:539
↓ 1 callersClassVisualTransformer
helpers/clip/core/clip.py:259
↓ 1 callersClassVoxelGrid
voxel/voxel_grid.py:12
ClassCNNAndFcsNet
helpers/network_utils.py:333
ClassCNNLangAndFcsNet
helpers/network_utils.py:393
ClassConv2DUpsampleBlock
helpers/network_utils.py:217
ClassConv3DInceptionBlockUpsampleBlock
helpers/network_utils.py:679
ClassConvTransposeUp3DBlock
helpers/network_utils.py:745
ClassCustomMultiTaskRLBenchEnv
helpers/custom_rlbench_env.py:206
ClassCustomMultiVariationRLBenchEnv
helpers/custom_rlbench_env.py:397
ClassCustomRLBenchEnv
helpers/custom_rlbench_env.py:20
ClassSimpleTokenizer
helpers/clip/core/simple_tokenizer.py:62
ClassViT
helpers/network_utils.py:555
ClassViTLangAndFcsNet
helpers/network_utils.py:595