Remove CPU optimization call and add logging for TPU strategy and data pipeline performance

model_training_nnn_tpu/trainer_tf.py

@@ -50,9 +50,6 @@ class BrainToTextDecoderTrainerTF:
         self.args = args
         self.logger = None
 
-        # Optimize CPU utilization for data pipeline (利用224核心)
-        self._configure_cpu_optimization()
-
         # Initialize TPU strategy
         self.strategy = create_tpu_strategy()
         if self.strategy is None:
@@ -60,6 +57,8 @@ class BrainToTextDecoderTrainerTF:
 
         print(f"Training on {self.strategy.num_replicas_in_sync} TPU cores")
         print(f"Strategy type: {type(self.strategy).__name__}")
+        print("💡 Using tf.data.AUTOTUNE for optimal data pipeline performance")
+        print("📝 Ensure create_input_fn uses AUTOTUNE for .map() and .prefetch() operations")
 
         # Configure mixed precision for TPU v5e-8
         if args.get('use_amp', True):