Aaditya1
/

Graphormer

Inference Endpoints

Model card Files Files and versions Community

Aaditya1 commited on Jul 22, 2023

Commit

3dbf6ec

•

1 Parent(s): 06379e5

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,           // Add the number of classes
   "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": 7,
   "pad_token_id": 0,
   "pre_layernorm": false,
   "q_noise": 0.0,