Unverified Commit 4d77d3fa authored by Jerry Zhang's avatar Jerry Zhang Committed by GitHub
Browse files

[fix] quantized googlenet (#1974)

https://github.com/pytorch/vision/pull/1949 seems to forget fixing quantized googlenet
parent 1f4f7624
...@@ -70,7 +70,8 @@ def googlenet(pretrained=False, progress=True, quantize=False, **kwargs): ...@@ -70,7 +70,8 @@ def googlenet(pretrained=False, progress=True, quantize=False, **kwargs):
if not original_aux_logits: if not original_aux_logits:
model.aux_logits = False model.aux_logits = False
del model.aux1, model.aux2 model.aux1 = None
model.aux2 = None
return model return model
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment