perf(cnn): 取消随机失活

zjZSTU · zjZSTU · commit 3e78f16af8cb · 2020-04-22T20:22:36.000+08:00
diff --git a/py/lib/models/yolo_v1.py b/py/lib/models/yolo_v1.py
@@ -56,7 +56,7 @@ def __init__(self, S=7, B=2, C=20):
         )
         self.avgpool = nn.AdaptiveAvgPool2d((7, 7))
         self.classifier = nn.Sequential(
-            nn.Dropout(),
+            # nn.Dropout(),
             nn.Linear(7 * 7 * 1024, 4096),
             nn.ReLU(inplace=True),
             nn.Linear(4096, S * S * (B * 5 + C))

Original file line number	Diff line number	Diff line change
`@@ -56,7 +56,7 @@ def __init__(self, S=7, B=2, C=20):`
`56`	`56`	`)`
`57`	`57`	`self.avgpool = nn.AdaptiveAvgPool2d((7, 7))`
`58`	`58`	`self.classifier = nn.Sequential(`
`59`		`- nn.Dropout(),`
	`59`	`+ # nn.Dropout(),`
`60`	`60`	`nn.Linear(7 * 7 * 1024, 4096),`
`61`	`61`	`nn.ReLU(inplace=True),`
`62`	`62`	`nn.Linear(4096, S * S * (B * 5 + C))`