ggt.models.ggt

Module Contents

Classes

GGT	Galaxy Group-Equivariant Transformer model.
GGTNoGConv	Galaxy Group-Equivariant Transformer model with no group

class ggt.models.ggt.GGT(cutout_size, channels, n_out=1, dropout=0.5)

Bases: torch.nn.Module

Galaxy Group-Equivariant Transformer model.

setup_stn(input_shape)

setup_featurizer()

setup_regression()

setup_pooling(input_shape=(6, 6))

setup_dropout(dropout)

spatial_transform(x)

forward(x)

class ggt.models.ggt.GGTNoGConv(cutout_size, channels, n_out=1, dropout=0.5)

Bases: GGT

Galaxy Group-Equivariant Transformer model with no group convolutional layers.

setup_featurizer()

forward(x)