Upload 2 files
Browse files- pretrain.pt +3 -0
- pretrain.yaml +8 -0
pretrain.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:39b463ebb02b97133521697425bd57d7905047a33a640eff84e1da048768cce3
|
| 3 |
+
size 55310054
|
pretrain.yaml
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
emb_dim: 300
|
| 2 |
+
model: gin-virtual
|
| 3 |
+
norm_layer: batch_norm
|
| 4 |
+
num_layer: 5
|
| 5 |
+
prior: 1.0e-09
|
| 6 |
+
readout: sum
|
| 7 |
+
threshold: 0.8
|
| 8 |
+
walk_length: 4
|