From 183b6a13ce0aeb14964e6fae11939dfb3d158bfe Mon Sep 17 00:00:00 2001 From: corochann Date: Mon, 9 Dec 2019 11:24:34 +0900 Subject: [PATCH] use n_layers=1 for readout --- chainer_chemistry/models/megnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)