Pytorch Patterns
Communauté/À propos de Pytorch Patterns
Cette compétence fournit des modèles idiomatiques PyTorch et les meilleures pratiques que l'assistant suit lorsqu'il aide avec le code d'apprentissage profond. Elle couvre le code indépendant de l'appareil, le seeding reproductible, la gestion des formes de tenseurs, l'architecture de modèle propre, l'initialisation appropriée des poids, les boucles d'entraînement et de validation, les pipelines de données, le checkpointing et les optimisations de performance telles que la précision mixte, le gradient checkpointing et torch.compile.
It also flags common anti-patterns, such as forgetting model.eval(), breaking autograd with in-place operations, and saving full models instead of state_dicts.
Fiche générée par IA à partir des informations publiques du projet. Elle peut contenir des erreurs : vérifiez la source d'origine.