MCPcopy Create free account

hub / github.com/bdsaglam/torch-scae / types & classes

Types & classes28 in github.com/bdsaglam/torch-scae

⨍ Functions 140◇ Types & classes 28
↓ 6 callersClassMAB
torch_scae/set_transformer.py:107
↓ 4 callersClassCapsuleLayer
Implementation of a capsule layer.
torch_scae/object_decoder.py:28
↓ 4 callersClassSetTransformer
Permutation-invariant Transformer.
torch_scae/set_transformer.py:174
↓ 3 callersClassCNNEncoder
torch_scae/part_encoder.py:26
↓ 3 callersClassCapsuleImageEncoder
torch_scae/part_encoder.py:47
↓ 3 callersClassCapsuleObjectDecoder
torch_scae/object_decoder.py:375
↓ 3 callersClassMultiHeadQKVAttention
Multi-head version of Transformer-like attention.
torch_scae/set_transformer.py:50
↓ 3 callersClassTemplateBasedImageDecoder
Template-based primary capsule decoder for images.
torch_scae/part_decoder.py:113
↓ 3 callersClassTemplateGenerator
Template-based primary capsule decoder for images.
torch_scae/part_decoder.py:31
↓ 2 callersClassCapsuleLikelihood
Capsule voting mechanism.
torch_scae/object_decoder.py:243
↓ 2 callersClassSCAE
Stacked Capsule Auto-Encoder
torch_scae/stacked_capsule_auto_encoder.py:22
↓ 1 callersClassISAB
torch_scae/set_transformer.py:145
↓ 1 callersClassLookAhead
LookAhead Optimizer Wrapper. Implementation modified from: https://github.com/rwightman/pytorch-image-models/blob/master/timm/optim/lookahead.py
torch_scae/optimizers.py:105
↓ 1 callersClassMNISTExperiment
torch_scae_experiments/mnist/experiment.py:22
↓ 1 callersClassPMA
torch_scae/set_transformer.py:161
↓ 1 callersClassRAdam
torch_scae/optimizers.py:10
↓ 1 callersClassSAB
torch_scae/set_transformer.py:136
ClassAttentionAveragedPooling2d
torch_scae/nn_ext.py:128
ClassBaseExperiment
torch_scae_experiments/base_experiment.py:32
ClassCapsuleImageEncoderTestCase
torch_scae/tests/test_part_encoder.py:8
ClassCapsuleLayerTestCase
torch_scae/tests/test_object_decoder.py:9
ClassCapsuleLikelihoodTestCase
torch_scae/tests/test_object_decoder.py:62
ClassCapsuleObjectDecoderTestCase
torch_scae/tests/test_object_decoder.py:115
ClassGaussianMixture
torch_scae/distributions.py:20
ClassSCAETestCase
torch_scae/tests/test_scae.py:15
ClassSetTransformerTestCase
torch_scae/tests/test_set_transformer.py:9
ClassTemplateBasedImageDecoderTestCase
torch_scae/tests/test_part_decoder.py:74
ClassTemplateGeneratorTestCase
torch_scae/tests/test_part_decoder.py:8