From d24edafb945507d9fbb2365e4d0dc58644f21ca3 Mon Sep 17 00:00:00 2001
From: polar1ce <951394033@qq.com>
Date: Tue, 16 Apr 2024 23:03:16 +0800
Subject: [PATCH] Update run.py

---
 run.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/run.py b/run.py
index 5d2a934..f05795f 100644
--- a/run.py
+++ b/run.py
@@ -12,8 +12,8 @@
 # 初始化模型和数据
 def initialize_model_and_data():
     INPUT_SIZE = 3
-    HIDDEN_SIZE = 10
-    NUM_LAYERS = 1
+    HIDDEN_SIZE = 32
+    NUM_LAYERS = 3
     OUTPUT_SIZE = 1
 
     lstm = LSTM(INPUT_SIZE, HIDDEN_SIZE, NUM_LAYERS, OUTPUT_SIZE)