pytorch-model-recovery by benchflow-ai

pytorch-model-recovery

benchflow-ai

9382423 months ago

Guidance for recovering PyTorch model architectures from state dictionaries, retraining specific layers, and saving models in TorchScript format. This skill should be used when tasks involve reconstructing model architectures from saved weights, fine-tuning specific layers while freezing others, or converting models to TorchScript format.

>_
Quick Install
npxskills add benchflow-ai/skillsbench--skill pytorch-model-recovery
Instructions

No instructions available