Burdenthrive commited on
Commit
f73e679
·
verified ·
1 Parent(s): dfa357b

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -10,7 +10,7 @@ class UNet(nn.Module):
10
 
11
  def __init__(self,
12
  encoder_name='tu-regnetz_d8',
13
- encoder_weights='imagenet', # Pre-trained weights from ImageNet
14
  in_channels=13, # Number of input channels (13 for Sentinel-2 multi-spectral images)
15
  num_classes=4, # Number of output classes (e.g., clear, thick cloud, thin cloud, cloud shadow)
16
  freeze_encoder=False): # Whether to freeze the encoder's weights
 
10
 
11
  def __init__(self,
12
  encoder_name='tu-regnetz_d8',
13
+ encoder_weights=None,
14
  in_channels=13, # Number of input channels (13 for Sentinel-2 multi-spectral images)
15
  num_classes=4, # Number of output classes (e.g., clear, thick cloud, thin cloud, cloud shadow)
16
  freeze_encoder=False): # Whether to freeze the encoder's weights