fix eval laoder on win (#2654)

This commit is contained in:
littletomatodonkey 2021-04-27 10:32:17 +08:00 committed by GitHub
parent c27022294e
commit 099957c907
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions

View File

@ -18,6 +18,7 @@ from __future__ import print_function
import os
import sys
import platform
import yaml
import time
import shutil
@ -338,8 +339,10 @@ def eval(model, valid_dataloader, post_process_class, eval_class,
total_frame = 0.0
total_time = 0.0
pbar = tqdm(total=len(valid_dataloader), desc='eval model:')
max_iter = len(valid_dataloader) - 1 if platform.system(
) == "Windows" else len(valid_dataloader)
for idx, batch in enumerate(valid_dataloader):
if idx >= len(valid_dataloader):
if idx >= max_iter:
break
images = batch[0]
start = time.time()