backup

2023-01-17 10:05:09 +00:00
parent b17baf157b
commit fd185d1b48
5 changed files with 222 additions and 91 deletions
@@ -244,14 +244,14 @@ class ConvNeXtReverseBase2(nn.Module):
        self.last = nn.Conv2d(
                    block_setting[-1].input_channels,
                    1,
-                    kernel_size=7,
-                    padding=3,
+                    kernel_size=3,
+                    padding='same',
                    ).to(device)
        # self.last = nn.Sequential(
        #                 Permute([0, 2, 3, 1]),
        #                 nn.Linear(in_features=block_setting[-1].input_channels, out_features=1, bias=True),
        #                 Permute([0, 3, 1, 2]),
-        #             )
+        #             ).to(device)


        for m in self.modules():