Skip to content

Commit 23e3bd3

Browse files
authored
support hybrid descriptor (fix #458) (#459)
1 parent 5c0cc0a commit 23e3bd3

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

dpgen/generator/run.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -376,7 +376,11 @@ def make_train (iter_index,
376376
# set random seed for each model
377377
if LooseVersion(mdata["deepmd_version"]) >= LooseVersion('1') and LooseVersion(mdata["deepmd_version"]) < LooseVersion('3'):
378378
# 1.x
379-
jinput['model']['descriptor']['seed'] = random.randrange(sys.maxsize) % (2**32)
379+
if jinput['model']['descriptor']['type'] == 'hybrid':
380+
for desc in jinput['model']['descriptor']['list']:
381+
desc['seed'] = random.randrange(sys.maxsize) % (2**32)
382+
else:
383+
jinput['model']['descriptor']['seed'] = random.randrange(sys.maxsize) % (2**32)
380384
jinput['model']['fitting_net']['seed'] = random.randrange(sys.maxsize) % (2**32)
381385
jinput['training']['seed'] = random.randrange(sys.maxsize) % (2**32)
382386
else:

0 commit comments

Comments
 (0)