galaxy.models.resnet18

Classes

Functions

load_model(in_channels, n_out_classes)

Module Contents

class galaxy.models.resnet18.ResNet18(in_channels: int, n_out_classes: int)

Bases: torch.nn.Module

resnet

forward(x)

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