Fix broken epoch calculation

jhonnold · jhonnold · commit f882ab2495e5 · 2025-03-05T05:28:26.000-07:00
diff --git a/src/models/chessmodel.h b/src/models/chessmodel.h
@@ -104,7 +104,7 @@ struct ChessModel : Model {
                 }
             }
 
-            float epoch_loss = total_epoch_loss / (val_epoch_size / train_loader.batch_size);
+            float epoch_loss = total_epoch_loss / (epoch_size / train_loader.batch_size);
             float val_loss   = (val_loader.has_value())
                                    ? total_val_loss / (val_epoch_size / val_loader->batch_size)
                                    : 0;

Original file line number	Diff line number	Diff line change
`@@ -104,7 +104,7 @@ struct ChessModel : Model {`
`104`	`104`	`}`
`105`	`105`	`}`
`106`	`106`
`107`		`- float epoch_loss = total_epoch_loss / (val_epoch_size / train_loader.batch_size);`
	`107`	`+ float epoch_loss = total_epoch_loss / (epoch_size / train_loader.batch_size);`
`108`	`108`	`float val_loss = (val_loader.has_value())`
`109`	`109`	`? total_val_loss / (val_epoch_size / val_loader->batch_size)`
`110`	`110`	`: 0;`