Update README.md
Browse files
README.md
CHANGED
|
@@ -16,4 +16,6 @@ The model is trained with the following configurations:
|
|
| 16 |
- **prior**: 1.0e-09
|
| 17 |
- **readout**: sum
|
| 18 |
- **threshold**: 0.8
|
| 19 |
-
- **walk_length**: 4
|
|
|
|
|
|
|
|
|
| 16 |
- **prior**: 1.0e-09
|
| 17 |
- **readout**: sum
|
| 18 |
- **threshold**: 0.8
|
| 19 |
+
- **walk_length**: 4
|
| 20 |
+
|
| 21 |
+
Please place the config.YAML file in the `ckpt` folder and rename it to `pretrain.yaml`.
|