This commit is contained in:
harisreedhar
2025-03-11 14:43:10 +01:00
committed by henryruhs
parent a3ac4d5ddd
commit 606bf42089
2 changed files with 2 additions and 1 deletions
+1 -1
View File
@@ -46,7 +46,7 @@ class MaskNet(nn.Module):
output_tensor = up_sample(output_tensor)
output_tensor = self.conv(output_tensor)
output_tensor = self.activation(output_tensor)
output_tensor = self.sigmoid(output_tensor)
return output_tensor