Fix messed up tests for dropout (pytorch#22893)

Summary:
Fix pytorch#22109.

I've confirmed with suo that this wasn't intentional.
Pull Request resolved: pytorch#22893

Differential Revision: D16288640

Pulled By: Chillee

fbshipit-source-id: 00fd6fe418ecefb304866a723051d0e5451ba4d5

test/test_jit.py

@@ -5960,7 +5960,6 @@ def __init__(self, in_channels, out_channels, **kwargs):
             @torch.jit.script_method
             def forward(self, x):
                 x = self.conv(x)
-                return x
                 x = self.bn(x)
                 return F.relu(x, inplace=True)
 
@@ -5972,7 +5971,6 @@ def __init__(self):
             @torch.jit.script_method
             def forward(self, x):
                 x = self.Conv2d_1a_3x3(x)
-                return x
                 return F.dropout(x, training=self.training)
 
         class EagerConv2d(torch.nn.Module):
@@ -5983,7 +5981,6 @@ def __init__(self, in_channels, out_channels, **kwargs):
 
             def forward(self, x):
                 x = self.conv(x)
-                return x
                 x = self.bn(x)
                 return F.relu(x, inplace=True)
 
@@ -5994,7 +5991,6 @@ def __init__(self):
 
             def forward(self, x):
                 x = self.Conv2d_1a_3x3(x)
-                return x
                 return F.dropout(x, training=self.training)
 
         script_input = torch.rand(4, 3, 299, 299)