Bug: Benchmarks - remove fp16 samples type converting time for cnn an…

…d lstm models (#330) **Description** Remove fp16 samples type converting time for cnn and lstm models.
microsoft · Mar 17, 2022 · 6e74918 · 6e74918
1 parent ff51a3c
commit 6e74918
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/superbench/benchmarks/model_benchmarks/pytorch_cnn.py b/superbench/benchmarks/model_benchmarks/pytorch_cnn.py
@@ -134,8 +134,8 @@ def _inference_step(self, precision):
             self._model.eval()
             while True:
                 for idx, sample in enumerate(self._dataloader):
-                    start = time.time()
                     sample = sample.to(dtype=getattr(torch, precision.value))
+                    start = time.time()
                     if self._gpu_available:
                         sample = sample.cuda()
                     self._model(sample)

diff --git a/superbench/benchmarks/model_benchmarks/pytorch_lstm.py b/superbench/benchmarks/model_benchmarks/pytorch_lstm.py
@@ -141,8 +141,8 @@ def _train_step(self, precision):
         curr_step = 0
         while True:
             for idx, sample in enumerate(self._dataloader):
-                start = time.time()
                 sample = sample.to(dtype=getattr(torch, precision.value))
+                start = time.time()
                 if self._gpu_available:
                     sample = sample.cuda()
                 self._optimizer.zero_grad()