use 0.001 by default for AdamW

OutisLi · OutisLi · commit 42f9eece0f38 · 2026-02-27T17:41:49.000+08:00
diff --git a/deepmd/utils/argcheck.py b/deepmd/utils/argcheck.py
@@ -2806,7 +2806,7 @@ def optimizer_adamw() -> list[Argument]:
             "weight_decay",
             float,
             optional=True,
-            default=0.0,
+            default=0.001,
             doc=doc_only_pt_supported + doc_weight_decay,
         ),
     ]
diff --git a/examples/water/dpa3/input_torch.json b/examples/water/dpa3/input_torch.json
@@ -68,6 +68,12 @@
     "limit_pref_v": 1,
     "_comment": " that's all"
   },
+  "optimizer": {
+    "type": "AdamW",
+    "adam_beta1": 0.9,
+    "adam_beta2": 0.999,
+    "weight_decay": 0.001
+  },
   "training": {
     "stat_file": "./dpa3.hdf5",
     "training_data": {
diff --git a/examples/water/dpa3/input_torch_dynamic.json b/examples/water/dpa3/input_torch_dynamic.json
@@ -70,6 +70,12 @@
     "limit_pref_v": 1,
     "_comment": " that's all"
   },
+  "optimizer": {
+    "type": "AdamW",
+    "adam_beta1": 0.9,
+    "adam_beta2": 0.999,
+    "weight_decay": 0.001
+  },
   "training": {
     "stat_file": "./dpa3.hdf5",
     "training_data": {

Original file line number	Diff line number	Diff line change
`@@ -2806,7 +2806,7 @@ def optimizer_adamw() -> list[Argument]:`
`2806`	`2806`	`"weight_decay",`
`2807`	`2807`	`float,`
`2808`	`2808`	`optional=True,`
`2809`		`- default=0.0,`
	`2809`	`+ default=0.001,`
`2810`	`2810`	`doc=doc_only_pt_supported + doc_weight_decay,`
`2811`	`2811`	`),`
`2812`	`2812`	`]`