diff --git a/chainer_chemistry/models/megnet.py b/chainer_chemistry/models/megnet.py index be8cf672..d56e7e66 100644 --- 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)