hikmatfarhat/MNIST_Classifier

1###import os,sys2###sys.path.insert(1,os.path.join(sys.path[0],".."))3from .network import Net4from .config import MNIST_config5from transformers import PreTrainedModel6# utils not used but importing it forces the upload to huggingface hub to include it7 8class MNIST_Classifier(PreTrainedModel):9    config_class = MNIST_config10    def __init__(self, config):11        super().__init__(config)12        self.classifier=Net(config.input_size,config.hidden_size1,config.hidden_size2,13           config.output_size)14 15    def forward(self, input):16        return self.classifier(input)17