feat: enhance AutoConfig to support path-based model configuration

Author: Tohrusky

cccv/arch/sr/dat_arch.py

@@ -365,8 +365,7 @@ def __init__(
         elif idx == 1:
             W_sp, H_sp = self.split_size[0], self.split_size[1]
         else:
-            print("[CCCV] ERROR MODE", idx)
-            exit(0)
+            raise ValueError(f"[CCCV] ERROR MODE: invalid idx {idx}, expected 0 or 1")
         self.H_sp = H_sp
         self.W_sp = W_sp
 

cccv/arch/sr/upcunet_arch.py

@@ -376,8 +376,7 @@ def forward(self, x, tile_mode, cache_mode, alpha, pro):
             t2 = tile_mode * 2
             crop_size = (((h0 - 1) // t2 * t2 + t2) // tile_mode, ((w0 - 1) // t2 * t2 + t2) // tile_mode)
         else:
-            print("[CCCV] tile_mode config error")
-            os._exit(233)
+            raise ValueError("[CCCV] tile_mode config error: invalid tile_mode value")
 
         ph = ((h0 - 1) // crop_size[0] + 1) * crop_size[0]
         pw = ((w0 - 1) // crop_size[1] + 1) * crop_size[1]
@@ -526,8 +525,7 @@ def forward_gap_sync(self, x, tile_mode, alpha, pro):
             t2 = tile_mode * 2
             crop_size = (((h0 - 1) // t2 * t2 + t2) // tile_mode, ((w0 - 1) // t2 * t2 + t2) // tile_mode)
         else:
-            print("[CCCV] tile_mode config error")
-            os._exit(233)
+            raise ValueError("[CCCV] tile_mode config error: invalid tile_mode value")
         ph = ((h0 - 1) // crop_size[0] + 1) * crop_size[0]
         pw = ((w0 - 1) // crop_size[1] + 1) * crop_size[1]
         x = F.pad(x, (18, 18 + pw - w0, 18, 18 + ph - h0), "reflect")
@@ -767,8 +765,7 @@ def forward(self, x, tile_mode, cache_mode, alpha, pro):
             t4 = tile_mode * 4
             crop_size = (((h0 - 1) // t4 * t4 + t4) // tile_mode, ((w0 - 1) // t4 * t4 + t4) // tile_mode)
         else:
-            print("[CCCV] tile_mode config error")
-            os._exit(233)
+            raise ValueError("[CCCV] tile_mode config error: invalid tile_mode value")
         ph = ((h0 - 1) // crop_size[0] + 1) * crop_size[0]
         pw = ((w0 - 1) // crop_size[1] + 1) * crop_size[1]
         x = F.pad(x, (14, 14 + pw - w0, 14, 14 + ph - h0), "reflect")
@@ -916,8 +913,7 @@ def forward_gap_sync(self, x, tile_mode, alpha, pro):
             t4 = tile_mode * 4
             crop_size = (((h0 - 1) // t4 * t4 + t4) // tile_mode, ((w0 - 1) // t4 * t4 + t4) // tile_mode)
         else:
-            print("[CCCV] tile_mode config error")
-            os._exit(233)
+            raise ValueError("[CCCV] tile_mode config error: invalid tile_mode value")
         ph = ((h0 - 1) // crop_size[0] + 1) * crop_size[0]
         pw = ((w0 - 1) // crop_size[1] + 1) * crop_size[1]
         x = F.pad(x, (14, 14 + pw - w0, 14, 14 + ph - h0), "reflect")
@@ -1162,8 +1158,7 @@ def forward(self, x, tile_mode, cache_mode, alpha, pro):
             t2 = tile_mode * 2
             crop_size = (((h0 - 1) // t2 * t2 + t2) // tile_mode, ((w0 - 1) // t2 * t2 + t2) // tile_mode)
         else:
-            print("[CCCV] tile_mode config error")
-            os._exit(233)
+            raise ValueError("[CCCV] tile_mode config error: invalid tile_mode value")
         ph = ((h0 - 1) // crop_size[0] + 1) * crop_size[0]
         pw = ((w0 - 1) // crop_size[1] + 1) * crop_size[1]
         x = F.pad(x, (19, 19 + pw - w0, 19, 19 + ph - h0), "reflect")
@@ -1323,8 +1318,7 @@ def forward_gap_sync(self, x, tile_mode, alpha, pro):
             t2 = tile_mode * 2
             crop_size = (((h0 - 1) // t2 * t2 + t2) // tile_mode, ((w0 - 1) // t2 * t2 + t2) // tile_mode)  # 5.6G
         else:
-            print("[CCCV] tile_mode config error")
-            os._exit(233)
+            raise ValueError("[CCCV] tile_mode config error: invalid tile_mode value")
         ph = ((h0 - 1) // crop_size[0] + 1) * crop_size[0]
         pw = ((w0 - 1) // crop_size[1] + 1) * crop_size[1]
         x = F.pad(x, (19, 19 + pw - w0, 19, 19 + ph - h0), "reflect")