search

NRCan / geo-deep-learning

Deep learning applied to georeferenced datasets
https://geo-deep-learning.readthedocs.io/en/latest/
MIT License
149 stars 49 forks source link

inference: add "total=" arg to tqdm to provide ETA during inference #512

Closed remtav closed 1 year ago

remtav commented 1 year ago

for sub_image, h_idxs, w_idxs, hann_win in tqdm(img_gen, position=0, leave=True, desc=f'Inferring on patches'):

would simply become:

for sub_image, h_idxs, w_idxs, hann_win in tqdm(img_gen, position=0, leave=True,
                desc=f'Inferring on patches', total=len(patch_list)):