galaxy.models.baseline

Classes

Functions

load_model(in_channels[, n_out_classes])

Module Contents

class galaxy.models.baseline.Baseline(in_channels, n_out_classes)

Bases: torch.nn.Module

conv1

bn1

conv2

bn2

conv3

bn3

conv4

bn4

global_avg_pool

fc

dropout

forward(x)

galaxy.models.baseline.load_model(in_channels: int, n_out_classes: int = 1)