Update config.json

config.json

@@ -1 +1,6 @@
-{"model_type": "simple_nn", "input_dim": 1, "hidden_dim": 16, "output_dim": 1}
+{
+  "architectures": ["SimpleNN"],
+  "model_type": "simple_nn",
+  "hidden_size": 16,
+  "num_labels": 1
+}