Étiquette : Modèles
-
Afficher un Résumé d’un Modèle PyTorch
Les modèles de deep learning développés en PyTorch peuvent être très grands et donc difficiles à appréhender. Il est alors tentant d’afficher un résumé du modèle avec lequel on travaille. Si, en Keras, l’on dispose de la méthode summary qui nous permet de faire model.summary() pour afficher un joli résumé de model, en PyTorch, nous…
-
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 