From 0c716a60b2742c70ad715ad4e9b23a3f4d035a77 Mon Sep 17 00:00:00 2001
From: Louis Jean <lsj456@gmail.com>
Date: Fri, 9 Apr 2021 14:51:42 +0200
Subject: [PATCH] fix(torch): fix very long ETA with iter_size != 1

---
 src/backends/torch/torchlib.cc | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/backends/torch/torchlib.cc b/src/backends/torch/torchlib.cc
index 69f43dafd..03623d60b 100644
--- 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)
                   {