mirror of https://github.com/tracel-ai/burn.git
Fix inverted epoch - iteration counts in valid progress (#1699)
This commit is contained in:
parent
b387829731
commit
b7ab19ac71
|
@ -95,8 +95,8 @@ pub fn run<B: AutodiffBackend>(device: B::Device) {
|
||||||
|
|
||||||
println!(
|
println!(
|
||||||
"[Valid - Epoch {} - Iteration {}] Loss {} | Accuracy {}",
|
"[Valid - Epoch {} - Iteration {}] Loss {} | Accuracy {}",
|
||||||
iteration,
|
|
||||||
epoch,
|
epoch,
|
||||||
|
iteration,
|
||||||
loss.clone().into_scalar(),
|
loss.clone().into_scalar(),
|
||||||
accuracy,
|
accuracy,
|
||||||
);
|
);
|
||||||
|
|
Loading…
Reference in New Issue