ggt.models.ggt_no_gcov

Module Contents

Classes

GGT_no_gconv

Galaxy Group-Equivariant Transformer model.

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

Bases: torch.nn.Module

Galaxy Group-Equivariant Transformer model.

spatial_transform(x)

forward(x)