Unverified Commit 3c47b050 authored by catboxanon's avatar catboxanon Committed by GitHub

Also make SwinIR skippable

parent f9935258
......@@ -145,11 +145,11 @@ def inference(img, model, tile, tile_overlap, window_size, scale):
with tqdm(total=len(h_idx_list) * len(w_idx_list), desc="SwinIR tiles") as pbar:
for h_idx in h_idx_list:
if state.interrupted:
if state.interrupted or state.skipped:
break
for w_idx in w_idx_list:
if state.interrupted:
if state.interrupted or state.skipped:
break
in_patch = img[..., h_idx: h_idx + tile, w_idx: w_idx + tile]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment