Skip to content

Commit f7f7f18

Browse files
authored
Chore (examples): add missing seed for descriptors in examples (#5098)
<!-- This is an auto-generated comment: release notes by coderabbit.ai --> ## Summary by CodeRabbit * **Chores** * Updated example configuration files to include configurable seed parameters across all model variants (DPA2, DPA3, SE-Atten, SE-E3-TEBD, multi-task learning), enhancing reproducibility and configuration flexibility. <sub>✏️ Tip: You can customize this high-level summary in your review settings.</sub> <!-- end of auto-generated comment: release notes by coderabbit.ai -->
1 parent b9c82d7 commit f7f7f18

11 files changed

Lines changed: 20 additions & 9 deletions

examples/water/dpa2/input_torch_compressible.json

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,8 @@
5353
"g1_out_mlp": true
5454
},
5555
"precision": "float64",
56-
"add_tebd_to_repinit_out": false
56+
"add_tebd_to_repinit_out": false,
57+
"seed": 1
5758
},
5859
"fitting_net": {
5960
"neuron": [

examples/water/dpa2/input_torch_large.json

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,8 @@
5252
"g1_out_mlp": true
5353
},
5454
"precision": "float64",
55-
"add_tebd_to_repinit_out": false
55+
"add_tebd_to_repinit_out": false,
56+
"seed": 1
5657
},
5758
"fitting_net": {
5859
"neuron": [

examples/water/dpa2/input_torch_medium.json

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,8 @@
5252
"g1_out_mlp": true
5353
},
5454
"precision": "float64",
55-
"add_tebd_to_repinit_out": false
55+
"add_tebd_to_repinit_out": false,
56+
"seed": 1
5657
},
5758
"fitting_net": {
5859
"neuron": [

examples/water/dpa2/input_torch_small.json

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,8 @@
5252
"g1_out_mlp": true
5353
},
5454
"precision": "float64",
55-
"add_tebd_to_repinit_out": false
55+
"add_tebd_to_repinit_out": false,
56+
"seed": 1
5657
},
5758
"fitting_net": {
5859
"neuron": [

examples/water/dpa3/input_torch.json

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,8 @@
3434
"activation_function": "silut:10.0",
3535
"use_tebd_bias": false,
3636
"precision": "float32",
37-
"concat_output_tebd": false
37+
"concat_output_tebd": false,
38+
"seed": 1
3839
},
3940
"fitting_net": {
4041
"neuron": [

examples/water/dpa3/input_torch_dynamic.json

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,8 @@
3636
"activation_function": "silut:10.0",
3737
"use_tebd_bias": false,
3838
"precision": "float32",
39-
"concat_output_tebd": false
39+
"concat_output_tebd": false,
40+
"seed": 1
4041
},
4142
"fitting_net": {
4243
"neuron": [

examples/water/se_atten/input_torch.json

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,8 @@
2525
"activation_function": "tanh",
2626
"scaling_factor": 1.0,
2727
"normalize": true,
28-
"temperature": 1.0
28+
"temperature": 1.0,
29+
"seed": 1
2930
},
3031
"fitting_net": {
3132
"neuron": [

examples/water/se_e3_tebd/input_torch.json

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,8 @@
1717
],
1818
"tebd_dim": 8,
1919
"tebd_input_mode": "concat",
20-
"activation_function": "tanh"
20+
"activation_function": "tanh",
21+
"seed": 1
2122
},
2223
"fitting_net": {
2324
"neuron": [

examples/water_multi_task/pytorch_example/input_torch.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,7 @@
5454
},
5555
"precision": "float64",
5656
"add_tebd_to_repinit_out": false,
57+
"seed": 1,
5758
"_comment": " that's all"
5859
},
5960
"_comment": "that's all"

examples/water_multi_task/pytorch_example/input_torch_sharefit.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,7 @@
5454
},
5555
"precision": "float64",
5656
"add_tebd_to_repinit_out": false,
57+
"seed": 1,
5758
"_comment": " that's all"
5859
},
5960
"shared_fit_with_id": {

0 commit comments

Comments
 (0)