galaxy.models.spinalnet_vgg

Classes

SpinalNet_VGG

Functions

load_model(in_channels, n_out_classes)

Module Contents

class galaxy.models.spinalnet_vgg.SpinalNet_VGG(num_ftrs, half_in_size, layer_width, n_out_classes=1)

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.spinalnet_vgg.load_model(in_channels: int, n_out_classes: int)