fix bug with 1-dim model_devi_activation_func (deepmodeling#484)

Cloudac7 · Dec 1, 2021 · 56df5e5 · 56df5e5
1 parent 2af33a8
commit 56df5e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dpgen/generator/run.py b/dpgen/generator/run.py
@@ -395,7 +395,7 @@ def make_train (iter_index,
                 jinput['model']['fitting_net']['activation_function'] = model_devi_activation_func[ii][1]
             if len(np.array(model_devi_activation_func).shape) == 1 :                                    # for backward compatibility, 1-dim list, not net-resolved
                 jinput['model']['descriptor']['activation_function'] = model_devi_activation_func[ii]
-                jinput['model']['descriptor']['activation_function'] = model_devi_activation_func[ii]
+                jinput['model']['fitting_net']['activation_function'] = model_devi_activation_func[ii]
         # dump the input.json
         with open(os.path.join(task_path, train_input_file), 'w') as outfile:
             json.dump(jinput, outfile, indent = 4)