Skip to content

Commit be18580

Browse files
authored
set the seed of type_embedding network (#689)
(cherry picked from commit 1f1a39c)
1 parent d30d557 commit be18580

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

dpgen/generator/run.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -394,6 +394,8 @@ def make_train (iter_index,
394394
else:
395395
jinput['model']['descriptor']['seed'] = random.randrange(sys.maxsize) % (2**32)
396396
jinput['model']['fitting_net']['seed'] = random.randrange(sys.maxsize) % (2**32)
397+
if 'type_embedding' in jinput['model']:
398+
jinput['model']['type_embedding']['seed'] = random.randrange(sys.maxsize) % (2**32)
397399
jinput['training']['seed'] = random.randrange(sys.maxsize) % (2**32)
398400
else:
399401
raise RuntimeError("DP-GEN currently only supports for DeePMD-kit 1.x or 2.x version!" )

0 commit comments

Comments
 (0)