{'input_size': 10000, 'hidden_size': 256, 'output_size': 2, 'model_type': 'mlp'}