use n_layers=1 for readout

chainer · Dec 9, 2019 · 183b6a1 · 183b6a1
1 parent 761de72
commit 183b6a1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chainer_chemistry/models/megnet.py b/chainer_chemistry/models/megnet.py
@@ -53,7 +53,7 @@ def __init__(self, out_dim=32, n_update_layers=3, dropout_ratio=-1,
                     skip_intermediate=(i == 0)
                 ) for i in range(n_update_layers)])
             self.readout = MEGNetReadout(out_dim=out_dim, in_channels=32,
-                                         n_layers=16, processing_steps=3,
+                                         n_layers=1, processing_steps=3,
                                          dropout_ratio=dropout_ratio,
                                          activation=activation)