`fedlab_core.models.lenet`¶

LeNet in PyTorch.

Module Contents¶

Classes¶

`Lenet`
`LeNet`

class fedlab_core.models.lenet.Lenet¶

Bases: torch.nn.Module

forward(self, x)¶

class fedlab_core.models.lenet.LeNet(out_dim=10, in_channel=1, img_sz=28)¶

Bases: torch.nn.Module

features(self, x)¶

logits(self, x)¶

forward(self, x)¶