Aaditya1
/

Graphormer

Model card Files Files and versions

Aaditya1 commited on Jul 24, 2023

Commit

1ae9b08

·

1 Parent(s): 1bfed02

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -52,7 +52,7 @@
   "num_out_degree": 512,
   "num_spatial": 512,
   "num_trans_layers_to_freeze": 0,
-  "num_classes": 7,
   "pad_token_id": 0,
   "pre_layernorm": false,
   "q_noise": 0.0,

   "num_out_degree": 512,
   "num_spatial": 512,
   "num_trans_layers_to_freeze": 0,
+  "num_classes": 1,
   "pad_token_id": 0,
   "pre_layernorm": false,
   "q_noise": 0.0,