net

Module Contents

LeNet

get_net(config)

class LeNet(in_channels=1, num_classes=10)

Bases: torch.nn.Module

forward(x)
get_net(config)