fix a bug in the number of frames for auto ratio (#1276)

deepmodeling · Jul 15, 2023 · a356e70 · a356e70
1 parent f2da7db
commit a356e70
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/dpgen/generator/run.py b/dpgen/generator/run.py
@@ -400,11 +400,11 @@ def make_train(iter_index, jdata, mdata):
  nframes += dpdata.LabeledSystem(
  sys_single, fmt="deepmd/npy"
  ).get_nframes()
-  if auto_ratio:
-  if ii == iter_index - 1:
-  number_new_frames += nframes
-  else:
-  number_old_frames += nframes
+ if auto_ratio:
+ if ii == iter_index - 1:
+ number_new_frames += nframes
+ else:
+ number_old_frames += nframes
  if nframes < fp_task_min:
  log_task(
  "nframes (%d) in data sys %s is too small, skip" % (nframes, jj)