Explorar o código

Removing superfluous return in UnetBlock forward

Jason Antic %!s(int64=6) %!d(string=hai) anos
pai
achega
42fd50368d
Modificáronse 1 ficheiros con 0 adicións e 1 borrados
  1. 0 1
      fasterai/modules.py

+ 0 - 1
fasterai/modules.py

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