agnapprox.nets.base.lenet5

Class definition for vanilla LeNet5 implementation

Module Contents

Classes

LeNet5

Defintion of vanilla LeNet5 architecture torch.nn.Module

class agnapprox.nets.base.lenet5.LeNet5(num_classes)

Bases: torch.nn.Module

Defintion of vanilla LeNet5 architecture torch.nn.Module

forward(features)