debug:tensorflow

Diagnose TensorFlow and Keras issues across model development, training, and deployment.

9|1|Updated Jan 12, 2026
One-click install
npx skills add https://github.com/SnakeO/claude-debug-and-refactor-skills-plugin --skill debug-tensorflow
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: debug:tensorflow
Source: https://github.com/SnakeO/claude-debug-and-refactor-skills-plugin/tree/main/plugins/debug-and-refactor/skills/debug-tensorflow
Command: npx skills add https://github.com/SnakeO/claude-debug-and-refactor-skills-plugin --skill debug-tensorflow

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

This Skill provides a structured approach to diagnosing and resolving TensorFlow and Keras issues that hinder model development, training stability, or deployment.

Core Features & Use Cases

  • Shape and dtype validation: quickly verify tensor shapes and data types match model expectations.
  • GPU/CUDA diagnostics and memory management: detect device availability, memory growth, and compatibility.
  • NaN/Inf loss handling: identify and mitigate numerical instability in loss and metrics.
  • Gradient analysis: inspect gradient flow and clipping to diagnose training stalls.
  • SavedModel loading and compatibility: ensure correct serialization formats and custom objects.
  • Data pipeline profiling: identify bottlenecks in tf.data pipelines and optimize throughput.

Quick Start

Run a minimal diagnostic on a TensorFlow project by enabling eager execution and performing a small-scale check on synthetic data and a tiny model.

Frequently Asked Questions about debug:tensorflow

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I fix TensorFlow GPU memory issues and CUDA errors?

To fix TensorFlow GPU memory issues, you need to diagnose device availability, enable memory growth, and verify CUDA compatibility. This skill systematically detects GPU/CUDA problems and optimizes memory management for training pipelines.

Why does my TensorFlow training loss return NaN or Inf values?

TensorFlow training loss returns NaN or Inf due to numerical instability in loss calculations or metrics. This skill identifies the source of NaN/Inf losses and provides mitigation strategies to stabilize model training.

How do I resolve tensor shape and dtype mismatches in Keras models?

Resolving tensor shape and dtype mismatches in Keras requires validating that tensor shapes and data types align with model expectations. This skill quickly verifies tensor configurations to prevent structural errors during model development.

What is the best way to identify bottlenecks in tf.data pipelines?

Identifying bottlenecks in tf.data pipelines involves profiling data throughput to isolate slow operations. This skill analyzes your data pipeline to pinpoint performance constraints and optimize end-to-end training throughput.

Can I diagnose TensorFlow gradient issues and training stalls?

You can diagnose TensorFlow gradient issues by inspecting gradient flow and applying clipping to address training stalls. This skill analyzes gradient behavior to help resolve stalled training sessions and improve convergence.

How do I fix SavedModel loading errors and custom object compatibility?

Fixing SavedModel loading errors requires ensuring correct serialization formats and proper handling of custom objects. This skill verifies compatibility to resolve deployment issues when loading TensorFlow models.