J'ai formé un modèle RESNET18 pré-formé en pytorch et l'ai sauvegardé. Tout en testant le modèle donne une précision différente pour la taille de mini-lot. Quelqu'un sait-il pourquoi?Pytorch Pré-formé RESNET18 Modèle
0
A
Répondre
0
Oui, je le pense. RESNET contient des couches de normalisation par lots. Au moment de l'évaluation, vous devez les corriger; sinon, les moyens de fonctionnement sont continuellement ajustés après le traitement de chaque lot, ce qui vous donne une précision différente.
réglage Essayez:
model.eval()
avant l'évaluation. Notez avant de reprendre la formation, appelez le model.train()
.
J'ai oublié de mettre à jour le fil avec la réponse. C'était précisément ce qui fonctionnait. Je vous remercie. – Snd