dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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

This commit is contained in:
Guillaume Lagrange 2024-04-26 15:26:09 -04:00 committed by GitHub
parent b387829731
commit b7ab19ac71
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/custom-training-loop/src/lib.rs
View File

@ -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,
);