tf-to-pytorch
CommunityStreamline TensorFlow to PyTorch model conversion for research.
Software Engineering#deep learning#pytorch#tensorflow#framework migration#model conversion#model verification
Authordongg622
Version1.0.0
Installs0
System Documentation
What problem does it solve?
This Skill facilitates converting TensorFlow or Keras models into PyTorch, enabling seamless migration between deep learning frameworks.
Core Features & Use Cases
- Model analysis: Examine layer structure and weight formats in TensorFlow models.
- Conversion guidance: Provide step-by-step instructions for mapping weights, layers, and operations from TF to PyTorch.
- Verification: Assist in validating numerical similarity layer-by-layer to ensure accurate translation.
- Use Case: Researchers can convert pre-trained TF models to PyTorch-compatible formats for deployment on Ascend NPUs or other platforms supporting PyTorch.
Quick Start
Analyze the TF model weights and layer outputs to guide conversion into PyTorch, then verify the consistency with numerical tests.
Dependency Matrix
Required Modules
tensorflowtorchnumpy
Components
scriptsreferences
💻 Claude Code Installation
Recommended: Let Claude install automatically. Simply copy and paste the text below to Claude Code.
Please help me install this Skill: Name: tf-to-pytorch Download link: https://github.com/dongg622/china-ai-chip-skill/archive/main.zip#tf-to-pytorch Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
Agent Skills Search Helper
Install a tiny helper to your Agent, search and equip skill from 471,000+ vetted skills library on demand.