galaxy.models.densenet

Classes

SpinalNet_DenseNet

Functions

load_model(in_channels, n_out_classes)

Module Contents

class galaxy.models.densenet.SpinalNet_DenseNet(num_ftrs, half_in_size, layer_width, n_out_classes)

Bases: torch.nn.Module

half_in_size

fc_spinal_layer1

fc_spinal_layer2

fc_spinal_layer3

fc_spinal_layer4

fc_out

forward(x)

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