We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 22bae94 commit bdcdc29Copy full SHA for bdcdc29
TrainingLoop/TrainingStatistics.swift
@@ -66,6 +66,8 @@ public class TrainingStatistics {
66
case .trainingStart, .validationStart:
67
totalBatchLoss = nil
68
totalBatches = nil
69
+ totalCorrect = nil
70
+ totalExamples = nil
71
case .batchEnd:
72
if metrics.contains(.accuracy) {
73
measureAccuracy(loop)
0 commit comments