cerebras.modelzoo.fc_mnist.pytorch.model.MNISTModel

class cerebras.modelzoo.fc_mnist.pytorch.model.MNISTModel(*args, **kwargs)

Bases: torch.nn.Module

Methods

build_model
forward

previous

cerebras.modelzoo.fc_mnist.pytorch.model.MNIST

next

cerebras.modelzoo.fc_mnist.pytorch.prepare_data