diff --git a/examples/custom-training-loop/src/lib.rs b/examples/custom-training-loop/src/lib.rs index 88d7dc646..89b9bd80f 100644 --- a/examples/custom-training-loop/src/lib.rs +++ b/examples/custom-training-loop/src/lib.rs @@ -95,8 +95,8 @@ pub fn run(device: B::Device) { println!( "[Valid - Epoch {} - Iteration {}] Loss {} | Accuracy {}", - iteration, epoch, + iteration, loss.clone().into_scalar(), accuracy, );