galaxy.models.vitL16

Classes

Functions

load_model(in_channels, n_out_classes)

Module Contents

class galaxy.models.vitL16.Sigm(n_out_classes=1)

Bases: torch.nn.Module

fc_out

forward(x)

galaxy.models.vitL16.load_model(in_channels: int, n_out_classes: int)