fix

Author: OutisLi

source/tests/tf/data_modifier/dipole.json

@@ -53,6 +53,10 @@
     "_comment4": " that's all"
   },
 
+  "optimizer": {
+    "type": "Adam"
+  },
+
   "training": {
     "training_data": {
       "systems": [
@@ -64,7 +68,6 @@
     "seed": 1,
     "disp_file": "lcurve.out",
     "disp_freq": 100,
-    "numb_test": 5,
     "save_freq": 500,
     "save_ckpt": "model.ckpt",
     "_comment8": "that's all"

source/tests/tf/test_data_modifier.py

@@ -67,7 +67,7 @@ def _setUp(self) -> None:
         systems = [tests_path / ii for ii in systems]
         set_pfx = "set"
         batch_size = jdata["training"]["training_data"]["batch_size"]
-        test_size = jdata["training"]["numb_test"]
+        test_size = 1
         data = DeepmdDataSystem(
             systems, batch_size, test_size, rcut, set_prefix=set_pfx
         )

source/tests/tf/test_data_modifier_shuffle.py

@@ -72,7 +72,7 @@ def _setUp(self) -> None:
         systems = jdata["training"]["training_data"]["systems"]
         set_pfx = "set"
         batch_size = jdata["training"]["training_data"]["batch_size"]
-        test_size = jdata["training"]["numb_test"]
+        test_size = 1
         data = DeepmdDataSystem(
             systems, batch_size, test_size, rcut, set_prefix=set_pfx
         )
@@ -177,13 +177,13 @@ def _setUp_jdata(self):
                 "decay_steps": 5000,
                 "decay_rate": 0.95,
             },
+            "optimizer": {"type": "Adam"},
             "training": {
                 "training_data": {
                     "systems": ["data_modifier/sys_test_0"],
                     "batch_size": 1,
                 },
                 "numb_steps": 1000000,
-                "numb_test": 2,
             },
         }
         return jdata