From 2453abe754d0e6c6cfc9682b28a3cc83ac1e0c4c Mon Sep 17 00:00:00 2001
From: Jay Paul Morgan <jaymiles17@gmail.com>
Date: Mon, 11 Dec 2023 11:42:04 +0000
Subject: [PATCH] Remove unused parameter
---
src/cloudremoval/model.py | 2 --
1 file changed, 2 deletions(-)
diff --git a/src/cloudremoval/model.py b/src/cloudremoval/model.py
index 8e1cc45..df22769 100644
--- a/src/cloudremoval/model.py
+++ b/src/cloudremoval/model.py
@@ -56,7 +56,6 @@ class CloudDivision(CloudIdentity):
class UNet(nn.Module):
def __init__(
self,
- n_blocks: int = 6,
cleaner=CloudAddition(),
in_channels=1,
out_channels=1,
@@ -65,7 +64,6 @@ class UNet(nn.Module):
super().__init__()
self.in_channels = in_channels
self.out_channels = out_channels
- self.n_blocks = n_blocks
self.cleaner = cleaner
self.features = init_features
--
GitLab