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

class cerebras.modelzoo.fc_mnist.pytorch.model.MNISTModel(*args, **kwargs)[source]#

Bases: torch.nn.Module

Methods

build_model

forward