Perform ReLU inplace

Author: adamjstewart

torchvision/models/unet.py

@@ -8,9 +8,9 @@
 def double_conv(in_channels, out_channels):
     return nn.Sequential(
         nn.Conv2d(in_channels, out_channels, 3),
-        nn.ReLU(),
+        nn.ReLU(inplace=True),
         nn.Conv2d(out_channels, out_channels, 3),
-        nn.ReLU()
+        nn.ReLU(inplace=True)
     )
 
 
@@ -52,7 +52,7 @@ def __init__(self, in_channels, out_channels):
         assert in_channels > out_channels
 
         self.upconv = nn.ConvTranspose2d(in_channels, out_channels, 2, 2)
-        self.relu = nn.ReLU()
+        self.relu = nn.ReLU(inplace=True)
         self.conv = double_conv(in_channels, out_channels)
 
     def forward(self, x, out):