File size: 486 Bytes
233e6e6 0bd2cee 233e6e6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from transformers import PretrainedConfig
class MNIST_config(PretrainedConfig):
model_type = "MNIST_Classifier"
def __init__(self, **kwargs):
super().__init__(**kwargs)
for key,value in kwargs.items():
setattr(self,key,value)
#print(key,value)
#self.input_size=kwargs['input_size']
#self.hidden_size1=kwargs["hidden_size1"]
#self.hidden_size2=kwargs["hidden_size2"]
#self.output_size=kwargs["output_size"]
|