fix eval laoder on win (#2654)
This commit is contained in:
parent
c27022294e
commit
099957c907
|
@ -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()
|
||||
|
|
Loading…
Reference in New Issue