modelzoo.fc_mnist.pytorch.model.MNISTModel

class modelzoo.fc_mnist.pytorch.model.MNISTModel

Bases: torch.nn.Module

Methods

build_model
forward

__call__(*args: Any, **kwargs: Any) → Any

Call self as a function.

__init__(params)

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

previous

modelzoo.fc_mnist.pytorch.model.MNIST

next

modelzoo.fc_mnist.pytorch.prepare_data