03070845

models_processing/model_koi/test.py

@@ -0,0 +1,10 @@
+#!/usr/bin/env python
+# -*- coding:utf-8 -*-
+# @FileName  :test.py
+# @Time      :2025/3/5 18:11
+# @Author    :David
+# @Company: shenyang JY
+
+import tensorflow as tf
+print("TensorFlow 版本:", tf.__version__)
+print("GPU 是否可用:", tf.config.list_physical_devices('GPU'))

models_processing/model_koi/tf_bp.py

@@ -40,6 +40,7 @@ class BPHandler(object):
         model = Sequential([
             Dense(64, input_dim=opt.Model['input_size'], activation='relu'),  # 输入层和隐藏层，10个神经元
             Dense(32, activation='relu'),  # 隐藏层，8个神经元
+            Dense(16, activation='relu'),  # 隐藏层，8个神经元
             Dense(1, activation='linear')  # 输出层，1个神经元（用于回归任务）
         ])
         adam = optimizers.Adam(learning_rate=opt.Model['learning_rate'], beta_1=0.9, beta_2=0.999, epsilon=1e-7, amsgrad=True)