Catégorie : Modèles
-
Compter le Nombre de Paramètres d’un Modèle PyTorch
Problème : Comment savoir quel est le nombre de paramètres d’un modèle PyTorch? Solution : Utiliser la ligne de code suivante: Compléments :
-
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…
-
model.eval() en PyTorch
Problème : À quoi sert model.eval() en PyTorch? Solution : model.eval() sert à désactiver certaines couches du modèle model lorsqu’on n’est plus en train de l’entraîner. En effet, certaines couches comme les couches de batch normalization ou de dropout sont souvent destinées uniquement à l’entraînement et peuvent générer des comportements non désirés lors de phases…
-
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 :
La Data Science 