bob.med.tb.models.alexnet¶

Functions

build_alexnet([pretrained])

Build Alexnet CNN

Classes

Alexnet([pretrained])

Alexnet module

class bob.med.tb.models.alexnet.Alexnet(pretrained=False)[source]¶

Bases: torch.nn.modules.module.Module

Alexnet module

Note: only usable with a normalized dataset

forward(x)[source]¶

Parameters: x (list) – list of tensors.
Returns: tensor
Return type: torch.Tensor

training: bool¶

bob.med.tb.models.alexnet.build_alexnet(pretrained=False)[source]¶

Build Alexnet CNN

Returns: module
Return type: torch.nn.Module