03111039

models_processing/model_koi/losses.py

@@ -56,8 +56,8 @@ class SouthLoss(Loss):
                  name: str = "south_loss",
                  reduction: str = "sum_over_batch_size"):
         # 参数校验
-        if not 0 <= cap <= 1:
-            raise ValueError("cap 必须为归一化后的值且位于 [0,1] 区间")
+        # if not 0 <= cap <= 1:
+        #     raise ValueError("cap 必须为归一化后的值且位于 [0,1] 区间")
 
         super().__init__(name=name, reduction=reduction)
 

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('./models_processing/model_koi/lstm.yaml', 'r', encoding='utf-8') as f:
+    with open('./lstm.yaml', 'r', encoding='utf-8') as f:
         args = yaml.safe_load(f)
 
     dh = DataHandler(logger, args)
@@ -41,6 +41,7 @@ def model_training_bp():
     # 获取程序开始时间
     start_time = time.time()
     result = {}
+    success = 0
     print("Program starts execution!")
     try:
         # ------------ 获取数据，预处理训练数据 ------------