Also make SwinIR skippable

2023-01-23 22:00:27 -05:00 · 2023-01-23 22:00:27 -05:00 · 3c47b05036
commit 3c47b05036
parent f993525820
1 changed files with 2 additions and 2 deletions
--- a/extensions-builtin/SwinIR/scripts/swinir_model.py
+++ b/extensions-builtin/SwinIR/scripts/swinir_model.py
@ -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]