Skip to content

Commit 5075b4d

Browse files
authored
Remove pretrained settings from model
1 parent e4b8356 commit 5075b4d

File tree

1 file changed

+1
-10
lines changed

1 file changed

+1
-10
lines changed

segmentation_models_pytorch/base/encoder_decoder.py

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -18,16 +18,7 @@ def __init__(self, encoder, decoder, activation):
1818
self.activation = nn.Sigmoid()
1919
else:
2020
raise ValueError('Activation should be "sigmoid" or "softmax"')
21-
22-
if encoder.pretrained:
23-
self.set_preprocessing_params(
24-
input_size=encoder.input_size,
25-
input_space=encoder.input_space,
26-
input_range=encoder.input_range,
27-
mean=encoder.mean,
28-
std=encoder.std,
29-
)
30-
21+
3122
def forward(self, x):
3223
x = self.encoder(x)
3324
x = self.decoder(x)

0 commit comments

Comments
 (0)