Catégorie : PyTorch
-
Créer un Modèle en PyTorch
Problème : Comment créer un modèle en PyTorch? Solution : Utiliser l’objet Module de torch.nn. Vous devez définir une classe qui hérite de Module et possède les deux méthodes suivantes: Voir ci-dessous un exemple générique. Exemple :
-
Créer un Dataset pour PyTorch
PyTorch fournit des librairies puissantes pour faire du deep learning avec Python. Pour des exercices pratiques, on peut utiliser des datasets déjà pré-définis. Néanmoins, il arrive que l’on veuille définir son propre dataset à utiliser avec PyTorch, notamment lorsque l’on travaille avec des données que l’on a collectées soi-même. Il faut alors voir comment définir…
-
Vérifier la disponibilité d’un GPU avec PyTorch
Lorsque l’on commence à utiliser un GPU pour faire des calculs en PyTorch, on fait déjà un premier pas en avant vers des calculs optimisés. Pour autant, comment vérifier si l’on dispose d’un GPU sur lequel passer nos tenseurs ? Heureusement, PyTorch dispose d’une fonction spécialement conçue pour vérifier la disponibilité d’un GPU. Problème :…
-
Mémoire Utilisée par mon GPU
Si vous rencontrez souvent une erreur de PyTorch qui vous indique que vous n’avez plus de place mémoire libre, vous aurez certainement envie de connaître l’espace mémoire disponible et utilisé par votre GPU. Mais comment faire? Je vous propose la solution suivante. La fonction mem_get_info de torch.cuda permet justement de récupérer un couple d’entiers donnant…
La Data Science 