update example e3

deepmodeling · Aug 14, 2024 · cac0daf · cac0daf
1 parent f4dd8c8
commit cac0daf
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/examples/e3/input_short_cpu.json b/examples/e3/input_short_cpu.json
@@ -14,7 +14,7 @@
             },
             "irreps_hidden": "32x0e+32x1o+16x2e",
             "n_layers": 3,
-            "avg_num_neighbors": 51,
+            "avg_num_neighbors": 86,
             "tp_radial_emb": true
         },
         "prediction":{

diff --git a/examples/e3/nnenv.ep1474.pth → examples/e3/ref_model/nnenv.ep1474.pth b/examples/e3/nnenv.ep1474.pth → examples/e3/ref_model/nnenv.ep1474.pth