From cac0dafbdbea24afa6b301618c495807989bca93 Mon Sep 17 00:00:00 2001
From: QG-phy <guqq_phy@qq.com>
Date: Thu, 15 Aug 2024 00:37:30 +0800
Subject: [PATCH] update example e3

---
 examples/e3/input_short_cpu.json             |   2 +-
 examples/e3/{ => ref_model}/nnenv.ep1474.pth | Bin
 2 files changed, 1 insertion(+), 1 deletion(-)
 rename examples/e3/{ => ref_model}/nnenv.ep1474.pth (100%)

diff --git a/examples/e3/input_short_cpu.json b/examples/e3/input_short_cpu.json
index ae0cf524..7184c19b 100644
--- 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 b/examples/e3/ref_model/nnenv.ep1474.pth
similarity index 100%
rename from examples/e3/nnenv.ep1474.pth
rename to examples/e3/ref_model/nnenv.ep1474.pth