Update networks

2024-06-22 15:01:02 +01:00 · 2024-06-22 15:01:02 +01:00 · b7755bfea8
commit b7755bfea8
parent 0e4fe7ab63
1 changed files with 173 additions and 144 deletions
--- a/src/editor/models/v1.py
+++ b/src/editor/models/v1.py
@ -4,7 +4,7 @@ import torch


 class HistogramRestorationNet(nn.Module):
-    def __init__(self, bin_count: int):
+    def __init__(self, **_):
        super(HistogramRestorationNet, self).__init__()

        self.conv1 = nn.Conv3d(in_channels=1, out_channels=16, kernel_size=3, padding=1)