Skip to content
Snippets Groups Projects
Commit 2453abe7 authored by Jay Paul Morgan's avatar Jay Paul Morgan
Browse files

Remove unused parameter

parent 5e4f86a9
No related branches found
No related tags found
No related merge requests found
Show whitespace changes
Inline Side-by-side
src/cloudremoval/model.py
+ 0
− 2
View file @ 2453abe7
...@@ -56,7 +56,6 @@ class CloudDivision(CloudIdentity): ...@@ -56,7 +56,6 @@ class CloudDivision(CloudIdentity):
class UNet(nn.Module): class UNet(nn.Module):
def __init__( def __init__(
self, self,
n_blocks: int = 6,
cleaner=CloudAddition(), cleaner=CloudAddition(),
in_channels=1, in_channels=1,
out_channels=1, out_channels=1,
...@@ -65,7 +64,6 @@ class UNet(nn.Module): ...@@ -65,7 +64,6 @@ class UNet(nn.Module):
super().__init__() super().__init__()
self.in_channels = in_channels self.in_channels = in_channels
self.out_channels = out_channels self.out_channels = out_channels
self.n_blocks = n_blocks
self.cleaner = cleaner self.cleaner = cleaner
self.features = init_features self.features = init_features
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment