catch more specific errors

2021-11-24 00:14:05 +08:00
parent 35ee6f781e
commit 7dd860a881
4 changed files with 4 additions and 4 deletions
--- a/realesrgan/data/realesrgan_dataset.py
+++ b/realesrgan/data/realesrgan_dataset.py
@@ -92,7 +92,7 @@ class RealESRGANDataset(data.Dataset):
        while retry > 0:
            try:
                img_bytes = self.file_client.get(gt_path, 'gt')
-            except Exception as e:
+            except (IOError, OSError) as e:
                logger = get_root_logger()
                logger.warn(f'File client error: {e}, remaining retry times: {retry - 1}')
                # change another file to read