aidotse/LeakPro

Loss in training loops

Ouverte

#239 ouverte le 24 févr. 2025

 (1 commentaire) (0 réaction) (0 personne assignée)Python (26 forks)auto 404
good first issue

Métriques du dépôt

Stars
 (23 étoiles)
Métriques de merge PR
 (Métriques PR en attente)

Description

Issue

Problem Description

The last batch of each epoch is usually smaller than the others. One approach is to sum the losses and divide by the total number of samples.

Currently, loss calculations in handlers sum the average loss of each batch and divide by the length of the dataloader.

Expected Behavior

What Needs to be Done

How Can It Be Tested or Reproduced

Guide contributeur