Fix inverted epoch - iteration counts in valid progress (tracel-ai#1699)

agelas · Apr 26, 2024 · b7ab19a · b7ab19a
1 parent b387829
commit b7ab19a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/custom-training-loop/src/lib.rs b/examples/custom-training-loop/src/lib.rs
@@ -95,8 +95,8 @@ pub fn run<B: AutodiffBackend>(device: B::Device) {
 
             println!(
                 "[Valid - Epoch {} - Iteration {}] Loss {} | Accuracy {}",
-                iteration,
                 epoch,
+                iteration,
                 loss.clone().into_scalar(),
                 accuracy,
             );