ggt.models.ggt_no_gcov

Classes

Galaxy Group-Equivariant Transformer model.

Module Contents

class ggt.models.ggt_no_gcov.GGT_no_gconv(cutout_size, channels, n_out=1)

Bases: torch.nn.Module

Galaxy Group-Equivariant Transformer model.

cutout_size

channels

expected_input_shape

n_out = 1

localization

ln_out_shape

fc_in_size

fc_loc

featurize

pool

regress

spatial_transform(x)

forward(x)