Salvando durante o treinamento

Marcado: 

Visualizando 2 posts - 1 até 2 (de 2 do total)
  • Autor
    Posts
  • #44475
    Rafael Barbosa
    Participante
      1. Quantas épocas são recomendas para treinar usando yolo v8?
      2. Há alguma forma de salvar os melhores pesos no google drive enquanto o sistema está realizando treinamento?
      #44488
      Gabriel Alves
      Moderador

        Olá Rafael!

        1. Para o YOLOv8 não há uma recomendação pois depende muito do dataset e de seu tamanho ou suas características, mas uma dica geral é começar com pelo menos uns 300 epochs. Em seguida, verifique se não ocorre o overfitting, caso não ocorra continue treinando por mais tempo. Aumente para 500, 700, 1000, e assim sucessivamente. Ou seja, enquanto não ocorrer o overfitting você pode continuar sempre treinando mais. E observe a métrica mAP durante o treinamento, quando notar que ela estabilizar e não aumentar mais após algumas épocas então treinou o suficiente (porém, oscilar um pouco entre as epochs é normal).
        2. Para salvar direto no Drive, você pode usar o parâmetro “project” da função train().
          Com ele, você pode mudar o local onde deseja que os pesos treinados sejam salvos. Se você passar um caminho que seja dentro de /content/gdrive/MyDrive/ (igual como foi demonstrado nas aulas de treinamento do YOLOv4) então os pesos atualizados serão salvos dentro do Drive. E os melhores pesos serão aqueles com “-best” no nome.

        Exemplo de código:

        project = ‘/content/gdrive/MyDrive/treinamentoYOLO’ # diretorio drive
        resultados = model.train(data=arquivo_config, epochs=10, imgsz=640, name=’yolov8s_modelo’, project=project)

      Visualizando 2 posts - 1 até 2 (de 2 do total)
      • Você deve fazer login para responder a este tópico.