Skip to content

Commit c9734d7

Browse files
committed
fix: skip setting seed when descriptor/fitting_net doesn't exist
1 parent 30bc1e5 commit c9734d7

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

dpgen/generator/run.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -583,7 +583,9 @@ def make_train_dp(iter_index, jdata, mdata):
583583
mdata["deepmd_version"]
584584
) < Version("3"):
585585
# 1.x
586-
if jinput["model"]["descriptor"]["type"] == "hybrid":
586+
if "descriptor" not in jinput["model"]:
587+
pass
588+
elif jinput["model"]["descriptor"]["type"] == "hybrid":
587589
for desc in jinput["model"]["descriptor"]["list"]:
588590
desc["seed"] = random.randrange(sys.maxsize) % (2**32)
589591
elif jinput["model"]["descriptor"]["type"] == "loc_frame":
@@ -592,9 +594,10 @@ def make_train_dp(iter_index, jdata, mdata):
592594
jinput["model"]["descriptor"]["seed"] = random.randrange(
593595
sys.maxsize
594596
) % (2**32)
595-
jinput["model"]["fitting_net"]["seed"] = random.randrange(sys.maxsize) % (
596-
2**32
597-
)
597+
if "fitting_net" in jinput["model"]:
598+
jinput["model"]["fitting_net"]["seed"] = random.randrange(
599+
sys.maxsize
600+
) % (2**32)
598601
if "type_embedding" in jinput["model"]:
599602
jinput["model"]["type_embedding"]["seed"] = random.randrange(
600603
sys.maxsize

0 commit comments

Comments
 (0)