File tree Expand file tree Collapse file tree 1 file changed +5
-1
lines changed
Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Original file line number Diff line number Diff 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 :
You can’t perform that action at this time.
0 commit comments