Skip to content

Commit 21512d3

Browse files
authored
fix bug with 1-dim model_devi_activation_func
1 parent 7f86e32 commit 21512d3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

dpgen/generator/run.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -395,7 +395,7 @@ def make_train (iter_index,
395395
jinput['model']['fitting_net']['activation_function'] = model_devi_activation_func[ii][1]
396396
if len(np.array(model_devi_activation_func).shape) == 1 : # for backward compatibility, 1-dim list, not net-resolved
397397
jinput['model']['descriptor']['activation_function'] = model_devi_activation_func[ii]
398-
jinput['model']['descriptor']['activation_function'] = model_devi_activation_func[ii]
398+
jinput['model']['fitting_net']['activation_function'] = model_devi_activation_func[ii]
399399
# dump the input.json
400400
with open(os.path.join(task_path, train_input_file), 'w') as outfile:
401401
json.dump(jinput, outfile, indent = 4)

0 commit comments

Comments
 (0)