Putting inplace on Dropout.

datumbox · datumbox · commit 25f8b26087a8 · 2020-12-17T18:20:36.000Z
diff --git a/torchvision/models/mobilenetv3.py b/torchvision/models/mobilenetv3.py
@@ -156,7 +156,7 @@ def __init__(
         self.classifier = nn.Sequential(
             nn.Linear(lastconv_output_channels, last_channel),
             nn.Hardswish(inplace=True),
-            nn.Dropout(p=0.2),
+            nn.Dropout(p=0.2, inplace=True),
             nn.Linear(last_channel, num_classes),
         )
 

Original file line number	Diff line number	Diff line change
`@@ -156,7 +156,7 @@ def __init__(`
`156`	`156`	`self.classifier = nn.Sequential(`
`157`	`157`	`nn.Linear(lastconv_output_channels, last_channel),`
`158`	`158`	`nn.Hardswish(inplace=True),`
`159`		`- nn.Dropout(p=0.2),`
	`159`	`+ nn.Dropout(p=0.2, inplace=True),`
`160`	`160`	`nn.Linear(last_channel, num_classes),`
`161`	`161`	`)`
`162`	`162`