This commit is contained in:
Zchen
2025-10-16 21:13:42 +08:00
parent a0b59c6987
commit dde6378481
2 changed files with 44 additions and 8 deletions

View File

@@ -832,7 +832,16 @@ def create_tpu_strategy():
print(" Common variables: COLAB_TPU_ADDR, TPU_NAME, TPU_WORKER_ID")
print("🔄 Falling back to default strategy (CPU/GPU)")
return tf.distribute.get_strategy()
fallback_strategy = tf.distribute.get_strategy()
print(f"🎯 Fallback strategy created: {type(fallback_strategy).__name__}")
print(f"📊 Fallback strategy replicas: {fallback_strategy.num_replicas_in_sync}")
# Ensure we never return None
if fallback_strategy is None:
print("⚠️ Warning: Default strategy is None, creating OneDeviceStrategy")
fallback_strategy = tf.distribute.OneDeviceStrategy("/CPU:0")
return fallback_strategy
def build_model_for_tpu(config):