Bug: Benchmarks - remove fp16 samples type converting time for cnn and lstm models (#330)

**Description**
Remove fp16  samples type converting time for cnn and lstm models.
This commit is contained in:
Yuting Jiang 2022-03-17 14:02:40 +08:00 коммит произвёл GitHub
Родитель ff51a3cee9
Коммит 6e74918044
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
2 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

@ -134,8 +134,8 @@ class PytorchCNN(PytorchBase):
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)

Просмотреть файл

@ -141,8 +141,8 @@ class PytorchLSTM(PytorchBase):
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()