set schnet default params
@@ -120,7 +120,7 @@ class SchNet(chainer.Chain):
and the result is concatenated
"""
- def __init__(self, hidden_dim=64, out_dim=1, n_layers=3,
+ def __init__(self, out_dim=1, hidden_dim=64, n_layers=3,
readout_hidden_dim=32, n_atom_types=MAX_ATOMIC_NUM,
concat_hidden=False):
super(SchNet, self).__init__()
