galaxy.models.alexnet_vgg

Classes

Functions

load_model(in_channels, n_out_classes)

Module Contents

class galaxy.models.alexnet_vgg.AlexNet_VGG(num_ftrs, n_out_classes)

Bases: torch.nn.Module

num_ftrs

classifier

forward(x)

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