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" },