Étiquette : Astuces PyTorch
-
Passer un Tenseur de 1 Channel à 3 Channels en PyTorch
Problème : Comment passer d’un tenseur à 1 channel à un tenseur à 3 channels en PyTorch? Solution : Certains modèles, notamment pour les images, prennent en entrée des tenseurs à 3 channels (images RGB). Hors, parfois, on veut les utiliser sur des images à 1 channel (images en nuances de gris). Pour ce faire,…
-
Geler les Paramètres de mon Modèle PyTorch
Problème : Comment geler les paramètres de mon modèle PyTorch? Solution : Il arrive parfois que l’on veuille geler les paramètres d’un modèle pour ne pas les mettre à jour lorsque l’on entraîne le modèle. Pour réaliser ce tour de passe-passe, il s’agit d’utiliser de mettre à False l’attribut requires_grad des paramètres du modèle. Regarder…
La Data Science 