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)