Transfer learning

Le transfert d'apprentissage réutilise un modèle pré-entraîné sur un grand jeu de données (par exemple ImageNet) pour une nouvelle tâche. On remplace la dernière couche et on entraîne seulement celle-ci, ou tout le réseau avec un petit taux d'apprentissage. Il permet de bons résultats avec peu de données.

for p in modele.parameters(): p.requires_grad = False
modele.fc = nn.Linear(512, 3)

Tout le glossaire