{"num_atom_features": 81, "num_bond_features": 4, "num_layer": 3, "dim": 128, "out_dim": 2}