galaxy.models.effnet

Classes

SpinalNet_EfficientNet

Functions

load_model(in_channels, n_out_classes)

Module Contents

class galaxy.models.effnet.SpinalNet_EfficientNet(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.effnet.load_model(in_channels: int, n_out_classes: int)