Referente ao curso Redes Neurais Convolucionais: Deep Learning com PyTorch, no capítulo Estratégias de Treino e atividade Extração de características
Por que você não utilizou a ativação SoftMax na ultima camada? já que é um problema de classificação.
Continue sua jornada tech com ainda mais tempo para aprender e evoluir