mirror of https://github.com/tracel-ai/burn.git
Switches epoch and iteration to be in the proper order for the custom training loop (#2171)
This commit is contained in:
parent
d4a1d2026d
commit
2755c36ed7
|
@ -78,8 +78,8 @@ pub fn run<B: AutodiffBackend>(device: B::Device) {
|
|||
|
||||
println!(
|
||||
"[Train - Epoch {} - Iteration {}] Loss {:.3} | Accuracy {:.3} %",
|
||||
iteration,
|
||||
epoch,
|
||||
iteration,
|
||||
loss.clone().into_scalar(),
|
||||
accuracy,
|
||||
);
|
||||
|
@ -104,8 +104,8 @@ pub fn run<B: AutodiffBackend>(device: B::Device) {
|
|||
|
||||
println!(
|
||||
"[Valid - Epoch {} - Iteration {}] Loss {} | Accuracy {}",
|
||||
iteration,
|
||||
epoch,
|
||||
iteration,
|
||||
loss.clone().into_scalar(),
|
||||
accuracy,
|
||||
);
|
||||
|
|
Loading…
Reference in New Issue