diff --git a/model_training_nnn_tpu/trainer_tf.py b/model_training_nnn_tpu/trainer_tf.py index 32feeb0..58e49ea 100644 --- a/model_training_nnn_tpu/trainer_tf.py +++ b/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):