Instructions to use Aaditya1/Graphormer with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Aaditya1/Graphormer with Transformers:
# Load model directly from transformers import GraphormerForGraphClassification model = GraphormerForGraphClassification.from_pretrained("Aaditya1/Graphormer", dtype="auto") - Notebooks
- Google Colab
- Kaggle
Update config.json
Browse files- config.json +1 -1
config.json
CHANGED
|
@@ -52,7 +52,7 @@
|
|
| 52 |
"num_out_degree": 512,
|
| 53 |
"num_spatial": 512,
|
| 54 |
"num_trans_layers_to_freeze": 0,
|
| 55 |
-
"num_classes":
|
| 56 |
"pad_token_id": 0,
|
| 57 |
"pre_layernorm": false,
|
| 58 |
"q_noise": 0.0,
|
|
|
|
| 52 |
"num_out_degree": 512,
|
| 53 |
"num_spatial": 512,
|
| 54 |
"num_trans_layers_to_freeze": 0,
|
| 55 |
+
"num_classes": 1,
|
| 56 |
"pad_token_id": 0,
|
| 57 |
"pre_layernorm": false,
|
| 58 |
"q_noise": 0.0,
|