Update beheaded_inception3.py

yhn112 · web-flow · commit 1cb357fac1be · 2020-05-08T06:00:39.000+03:00
diff --git a/homework04/beheaded_inception3.py b/homework04/beheaded_inception3.py
@@ -43,7 +43,7 @@ def forward(self, x):
     
 
 def beheaded_inception_v3(transform_input=True):
-    model= BeheadedInception3(transform_input=transform_input)
+    model= BeheadedInception3(transform_input=transform_input, init_weights=False)
     inception_url = 'https://download.pytorch.org/models/inception_v3_google-1a9a5a14.pth'
     model.load_state_dict(load_url(inception_url))
     return model
