Melhor época de treinamento do modelo

[Brenda Mendes]

Como saber qual foi a melhor época do modelo treinado? Após cada treinamento retorna os gráficos e resultados daquele treinamento considerando todas as épocas, mas tem uma forma de saber qual foi a melhor época de todo o treinamento realizado no modelo e ter acesso a esses resultados em específico?

[Gabriel Alves]

Olá Brenda!

Para saber qual a época que corresponde ao melhor modelo treinado (o best.pt) você pode consultar o arquivo results.csv que é executado durante o treinamento. Este arquivo registra métricas como precisão, recall, mAP e perda de valor para cada época. Procure a época com melhor desempenho (geralmente a menor perda de validação ou o maior mAP, dependendo de qual best.pt foi selecionado) para identificar a época a que corresponde o seu modelo best.pt. Lembrando que esse arquivo best.pt (contendo os pesos salvos para o melhor resultado) também está acessível dentro do diretório de treinamento, junto ao last.pt (os pesos mais recentes salvos, independente de serem os melhores ou não).

[Author]

Posts