Update configuration_graphormer.pyx
Browse files
configuration_graphormer.pyx
CHANGED
|
@@ -132,7 +132,7 @@ class GraphormerConfig(PretrainedConfig):
|
|
| 132 |
|
| 133 |
def __init__(
|
| 134 |
self,
|
| 135 |
-
num_classes: int =
|
| 136 |
num_atoms: int = 512 * 9,
|
| 137 |
num_edges: int = 512 * 3,
|
| 138 |
num_in_degree: int = 512,
|
|
|
|
| 132 |
|
| 133 |
def __init__(
|
| 134 |
self,
|
| 135 |
+
num_classes: int = 7,
|
| 136 |
num_atoms: int = 512 * 9,
|
| 137 |
num_edges: int = 512 * 3,
|
| 138 |
num_in_degree: int = 512,
|