docsexp.models.cnn

Module Contents

Classes

ConvNet

Convolutional network

class docsexp.models.cnn.ConvNet(num_classes: int = 10)

Bases: torch.nn.Module

Convolutional network

Parameters

num_classes (int) – number of output classes

Example

>>> from docsexp.models import ConvNet
>>> num_classes = 10
>>> model = ConvNet(num_classes)

forward(self, x)

Forward function for CNN.

Parameters

x (torch.Tensor) – input features