diff --git a/doc/use-deepmd-kit.md b/doc/use-deepmd-kit.md index cbf7a91cbc..a6e3915b3b 100644 --- a/doc/use-deepmd-kit.md +++ b/doc/use-deepmd-kit.md @@ -156,7 +156,7 @@ An example of `training` is "numb_test": 10, "save_freq": 1000, "save_ckpt": "model.ckpt", - "load_ckpt": "model.ckpt", + "disp_training":true, "time_training":true, "profiling": false, @@ -189,7 +189,7 @@ During the training, the error of the model is tested every **`disp_freq`** batc ``` The first column displays the number of batches. The second and third columns display the loss function evaluated by `numb_test` frames randomly chosen from the test set and that evaluated by the current training batch, respectively. The fourth and fifth columns display the RMS energy error (normalized by number of atoms) evaluated by `numb_test` frames randomly chosen from the test set and that evaluated by the current training batch, respectively. The sixth and seventh columns display the RMS force error (component-wise) evaluated by `numb_test` frames randomly chosen from the test set and that evaluated by the current training batch, respectively. The last column displays the current learning rate. -Checkpoints will be written to files with prefix **`save_ckpt`** every **`save_freq`** batches. If **`restart`** is set to `true`, then the training will start from the checkpoint named **`load_ckpt`**, rather than from scratch. +Checkpoints will be written to files with prefix **`save_ckpt`** every **`save_freq`** batches. Several command line options can be passed to `dp train`, which can be checked with ```bash diff --git a/examples/fparam/train/input.json b/examples/fparam/train/input.json index c57afdfb7f..0455f11b42 100644 --- a/examples/fparam/train/input.json +++ b/examples/fparam/train/input.json @@ -51,7 +51,7 @@ "numb_test": 10, "save_freq": 1000, "save_ckpt": "model.ckpt", - "load_ckpt": "model.ckpt", + "disp_training":true, "time_training":true, "profiling": false, diff --git a/examples/fparam/train/input_aparam.json b/examples/fparam/train/input_aparam.json index 86be27ef29..5774130fe1 100644 --- a/examples/fparam/train/input_aparam.json +++ b/examples/fparam/train/input_aparam.json @@ -51,7 +51,7 @@ "numb_test": 10, "save_freq": 1000, "save_ckpt": "model.ckpt", - "load_ckpt": "model.ckpt", + "disp_training":true, "time_training":true, "profiling": false, diff --git a/examples/water/train/polar.json b/examples/water/train/polar.json index 60e3fa3494..3437747e49 100644 --- a/examples/water/train/polar.json +++ b/examples/water/train/polar.json @@ -53,7 +53,7 @@ "numb_test": 10, "save_freq": 1000, "save_ckpt": "model.ckpt", - "load_ckpt": "model.ckpt", + "disp_training":true, "time_training":true, "_comment": "that's all" diff --git a/examples/water/train/polar_se_a.json b/examples/water/train/polar_se_a.json index dc90e481ce..e0c2939722 100644 --- a/examples/water/train/polar_se_a.json +++ b/examples/water/train/polar_se_a.json @@ -51,7 +51,7 @@ "numb_test": 10, "save_freq": 1000, "save_ckpt": "model.ckpt", - "load_ckpt": "model.ckpt", + "disp_training":true, "time_training":true, "_comment": "that's all" diff --git a/examples/water/train/wannier.json b/examples/water/train/wannier.json index f23f5e0d62..06b3bc906d 100644 --- a/examples/water/train/wannier.json +++ b/examples/water/train/wannier.json @@ -54,7 +54,7 @@ "numb_test": 10, "save_freq": 1000, "save_ckpt": "model.ckpt", - "load_ckpt": "model.ckpt", + "disp_training":true, "time_training":true, "_comment": "that's all" diff --git a/examples/water/train/water.json b/examples/water/train/water.json index 23ba559aed..ccc47a5d3a 100644 --- a/examples/water/train/water.json +++ b/examples/water/train/water.json @@ -61,7 +61,7 @@ "numb_test": 10, "save_freq": 1000, "save_ckpt": "model.ckpt", - "load_ckpt": "model.ckpt", + "disp_training":true, "time_training":true, "_comment": "that's all" diff --git a/examples/water/train/water_se_a.json b/examples/water/train/water_se_a.json index cb005530c1..5c9b2d8e70 100644 --- a/examples/water/train/water_se_a.json +++ b/examples/water/train/water_se_a.json @@ -56,7 +56,7 @@ "numb_test": 10, "save_freq": 1000, "save_ckpt": "model.ckpt", - "load_ckpt": "model.ckpt", + "disp_training":true, "time_training":true, "profiling": false, diff --git a/examples/water/train/water_se_ar.json b/examples/water/train/water_se_ar.json index 2173f2e1d9..c97b3d1f70 100644 --- a/examples/water/train/water_se_ar.json +++ b/examples/water/train/water_se_ar.json @@ -67,7 +67,7 @@ "numb_test": 10, "save_freq": 1000, "save_ckpt": "model.ckpt", - "load_ckpt": "model.ckpt", + "disp_training":true, "time_training":true, "profiling": false, diff --git a/examples/water/train/water_se_r.json b/examples/water/train/water_se_r.json index 7faf55a3c3..3954e0b9aa 100644 --- a/examples/water/train/water_se_r.json +++ b/examples/water/train/water_se_r.json @@ -56,7 +56,7 @@ "numb_test": 10, "save_freq": 1000, "save_ckpt": "model.ckpt", - "load_ckpt": "model.ckpt", + "disp_training":true, "time_training":true, "profiling": false, diff --git a/examples/water/train/water_srtab_example.json b/examples/water/train/water_srtab_example.json index f2a0a4a39c..3baabd44fd 100644 --- a/examples/water/train/water_srtab_example.json +++ b/examples/water/train/water_srtab_example.json @@ -64,7 +64,7 @@ "numb_test": 10, "save_freq": 1000, "save_ckpt": "model.ckpt", - "load_ckpt": "model.ckpt", + "disp_training":true, "time_training":true, "_comment": "that's all"