Эх сурвалжийг харах

Removing superfluous return in UnetBlock forward

Jason Antic 6 жил өмнө
parent
commit
42fd50368d

+ 0 - 1
fasterai/modules.py

@@ -95,7 +95,6 @@ class UnetBlock(nn.Module):
         x = torch.cat([up_p,x_p], dim=1)
         x = torch.cat([up_p,x_p], dim=1)
         x = self.relu(x)
         x = self.relu(x)
         return self.out(x)
         return self.out(x)
-        return out
 
 
 class SaveFeatures():
 class SaveFeatures():
     features=None
     features=None