Update inceptionv4.py

ekagra-ranjan · web-flow · commit e07fb68c3178 · 2019-01-20T13:05:14.000+05:30
diff --git a/pretrainedmodels/models/inceptionv4.py b/pretrainedmodels/models/inceptionv4.py
@@ -295,7 +295,6 @@ def __init__(self, num_classes=1001):
             Inception_C(),
             Inception_C()
         )
-        self.avg_pool = nn.AvgPool2d(8, count_include_pad=False)
         self.last_linear = nn.Linear(1536, num_classes)
 
     def logits(self, features):

Original file line number	Diff line number	Diff line change
`@@ -295,7 +295,6 @@ def __init__(self, num_classes=1001):`
`295`	`295`	`Inception_C(),`
`296`	`296`	`Inception_C()`
`297`	`297`	`)`
`298`		`- self.avg_pool = nn.AvgPool2d(8, count_include_pad=False)`
`299`	`298`	`self.last_linear = nn.Linear(1536, num_classes)`
`300`	`299`
`301`	`300`	`def logits(self, features):`