David 3 ماه پیش
والد
کامیت
1a5748a189
1فایلهای تغییر یافته به همراه1 افزوده شده و 1 حذف شده
  1. 1 1
      models_processing/model_koi/tf_lstm_train.py

+ 1 - 1
models_processing/model_koi/tf_lstm_train.py

@@ -19,7 +19,7 @@ np.random.seed(42)  # NumPy随机种子
 app = Flask('tf_lstm_train——service')
 
 with app.app_context():
-    with open('./lstm.yaml', 'r', encoding='utf-8') as f:
+    with open('./models_processing/model_koi/lstm.yaml', 'r', encoding='utf-8') as f:
         args = yaml.safe_load(f)
 
     dh = DataHandler(logger, args)