docs: set precision explicitly in the DPA-2 example

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Nov 17, 2024 · 04c883a · 04c883a
1 parent 0ad4289
commit 04c883a
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/examples/water/dpa2/input_torch_large.json b/examples/water/dpa2/input_torch_large.json
@@ -51,6 +51,7 @@
         "g1_out_conv": true,
         "g1_out_mlp": true
       },
+      "precision": "float64",
       "add_tebd_to_repinit_out": false
     },
     "fitting_net": {
@@ -60,6 +61,7 @@
         240
       ],
       "resnet_dt": true,
+      "precision": "float64",
       "seed": 1,
       "_comment": " that's all"
     },

diff --git a/examples/water/dpa2/input_torch_medium.json b/examples/water/dpa2/input_torch_medium.json
@@ -51,6 +51,7 @@
         "g1_out_conv": true,
         "g1_out_mlp": true
       },
+      "precision": "float64",
       "add_tebd_to_repinit_out": false
     },
     "fitting_net": {
@@ -60,6 +61,7 @@
         240
       ],
       "resnet_dt": true,
+      "precision": "float64",
       "seed": 1,
       "_comment": " that's all"
     },

diff --git a/examples/water/dpa2/input_torch_small.json b/examples/water/dpa2/input_torch_small.json
@@ -51,6 +51,7 @@
         "g1_out_conv": true,
         "g1_out_mlp": true
       },
+      "precision": "float64",
       "add_tebd_to_repinit_out": false
     },
     "fitting_net": {
@@ -60,6 +61,7 @@
         240
       ],
       "resnet_dt": true,
+      "precision": "float64",
       "seed": 1,
       "_comment": " that's all"
     },