cerebras.modelzoo.fc_mnist.pytorch.model.MNISTModel#

class cerebras.modelzoo.fc_mnist.pytorch.model.MNISTModel[source]#

Bases: torch.nn.Module

Methods

`build_model`
`forward`

__init__(params)[source]#

__call__(*args: Any, **kwargs: Any) → Any#: Call self as a function.

static __new__(cls, *args: Any, **kwargs: Any) → Any#

previous

cerebras.modelzoo.fc_mnist.pytorch.model.MNIST

next

cerebras.modelzoo.fc_mnist.pytorch.prepare_data