fix(torch): fix very long ETA with iter_size != 1

jolibrain · Apr 14, 2021 · 0c716a6 · 0c716a6
1 parent e674fcc
commit 0c716a6
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/backends/torch/torchlib.cc b/src/backends/torch/torchlib.cc
@@ -817,8 +817,7 @@ namespace dd
                     += duration_cast<milliseconds>(tstop - tstart).count();
                 this->add_meas("iteration_duration_ms", last_it_time);
 
-                double remain_time_ms
-                    = last_it_time * iter_size * (iterations - it);
+                double remain_time_ms = last_it_time * (iterations - it);
 
                 if (test_interval > 0)
                   {