new model

img_demoAE.py

@@ -15,7 +15,8 @@ print(f'loading...')
 ########################################################################################################
 
 # model_prefix = 'out-v7c_d8_256-224-13bit-OB32x0.5-745'
-model_prefix = 'out-v7d_d16_512-224-13bit-OB32x0.5-2487'
+# model_prefix = 'out-v7d_d16_512-224-13bit-OB32x0.5-2487'
+model_prefix = 'out-v7d_d32_1024-224-13bit-OB32x0.5-3642'
 input_imgs = ['lena.png', 'genshin.png', 'kodim14-modified.png', 'kodim19-modified.png', 'kodim24-modified.png']
 device = 'cpu' # cpu cuda
 
@@ -154,7 +155,7 @@ else:
             self.args = args
             if 'd16_512' in model_prefix:
                 dd, ee, ff = 16, 64, 512
-            else:
+            elif 'd32_1024' in model_prefix:
                 dd, ee, ff = 32, 128, 1024
             self.CXX = nn.Conv2d(3, dd, kernel_size=3, padding=1)
             self.BXX = nn.BatchNorm2d(dd)
@@ -186,7 +187,7 @@ else:
             self.args = args
             if 'd16_512' in model_prefix:
                 dd, ee, ff = 16, 64, 512
-            else:
+            elif 'd32_1024' in model_prefix:
                 dd, ee, ff = 32, 128, 1024
             self.CZZ = nn.Conv2d(args.my_img_bit, dd*64, kernel_size=3, padding=1)
             self.BZZ = nn.BatchNorm2d(dd*64)

out-v7d_d32_1024-224-13bit-OB32x0.5-3642-D.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c22998d32e1ca754416504823ab72393a3888c48c9695493e6e8cf577a992434
+size 397763529

out-v7d_d32_1024-224-13bit-OB32x0.5-3642-E.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5ee11f299c5afb94de17dc598a76d90addf458cbcc63b8dcf4bec8dc7b4d99dc
+size 397723209