liuganghuggingface
/

InfoAlign-Pretrained

Graph Machine Learning

Model card Files Files and versions Community

liuganghuggingface commited on Jul 2

Commit

96977bb

•

1 Parent(s): 34bdf30

Update README.md

Files changed (1) hide show

README.md +3 -1

README.md CHANGED Viewed

@@ -16,4 +16,6 @@ The model is trained with the following configurations:
 - **prior**: 1.0e-09
 - **readout**: sum
 - **threshold**: 0.8
-- **walk_length**: 4

 - **prior**: 1.0e-09
 - **readout**: sum
 - **threshold**: 0.8
+- **walk_length**: 4
+Please place the config.YAML file in the `ckpt` folder and rename it to `pretrain.yaml`.