From 04c883a38689a9776c542eb885929ea39590751c Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Sun, 17 Nov 2024 18:28:19 -0500 Subject: [PATCH] docs: set precision explicitly in the DPA-2 example Signed-off-by: Jinzhe Zeng --- examples/water/dpa2/input_torch_large.json | 2 ++ examples/water/dpa2/input_torch_medium.json | 2 ++ examples/water/dpa2/input_torch_small.json | 2 ++ 3 files changed, 6 insertions(+) diff --git a/examples/water/dpa2/input_torch_large.json b/examples/water/dpa2/input_torch_large.json index 99eb14ec15..49edc2721e 100644 --- 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 index 804fb4ad68..0ba83f2cd5 100644 --- 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 index bf506dcb5f..86217d980e 100644 --- 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" },