diff options
| author | catboxanon <122327233+catboxanon@users.noreply.github.com> | 2023-01-23 22:00:27 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-01-23 22:00:27 -0500 |
| commit | 3c47b050367ee220dcfed7be7883878417735614 (patch) | |
| tree | 9aa7671d942a6aa0e5315fb922b68a3d8aa97cdd /extensions-builtin/SwinIR/scripts | |
| parent | f99352582084890b9167c1bf8699865bea0cef5f (diff) | |
Also make SwinIR skippable
Diffstat (limited to 'extensions-builtin/SwinIR/scripts')
| -rw-r--r-- | extensions-builtin/SwinIR/scripts/swinir_model.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/extensions-builtin/SwinIR/scripts/swinir_model.py b/extensions-builtin/SwinIR/scripts/swinir_model.py index 3479760a..e8783bca 100644 --- 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] |