Commit c94a1585 authored by taylanbil's avatar taylanbil Committed by Soumith Chintala
Browse files

typo fix: ouput -> output (#1034)

I grepped the repo for Ouputs and these were the only occurences
parent fefe118c
...@@ -12,7 +12,7 @@ model_urls = { ...@@ -12,7 +12,7 @@ model_urls = {
'googlenet': 'https://download.pytorch.org/models/googlenet-1378be20.pth', 'googlenet': 'https://download.pytorch.org/models/googlenet-1378be20.pth',
} }
_GoogLeNetOuputs = namedtuple('GoogLeNetOuputs', ['logits', 'aux_logits2', 'aux_logits1']) _GoogLeNetOutputs = namedtuple('GoogLeNetOutputs', ['logits', 'aux_logits2', 'aux_logits1'])
def googlenet(pretrained=False, progress=True, **kwargs): def googlenet(pretrained=False, progress=True, **kwargs):
...@@ -157,7 +157,7 @@ class GoogLeNet(nn.Module): ...@@ -157,7 +157,7 @@ class GoogLeNet(nn.Module):
x = self.fc(x) x = self.fc(x)
# N x 1000 (num_classes) # N x 1000 (num_classes)
if self.training and self.aux_logits: if self.training and self.aux_logits:
return _GoogLeNetOuputs(x, aux2, aux1) return _GoogLeNetOutputs(x, aux2, aux1)
return x return x
......
...@@ -13,7 +13,7 @@ model_urls = { ...@@ -13,7 +13,7 @@ model_urls = {
'inception_v3_google': 'https://download.pytorch.org/models/inception_v3_google-1a9a5a14.pth', 'inception_v3_google': 'https://download.pytorch.org/models/inception_v3_google-1a9a5a14.pth',
} }
_InceptionOuputs = namedtuple('InceptionOuputs', ['logits', 'aux_logits']) _InceptionOutputs = namedtuple('InceptionOutputs', ['logits', 'aux_logits'])
def inception_v3(pretrained=False, progress=True, **kwargs): def inception_v3(pretrained=False, progress=True, **kwargs):
...@@ -147,7 +147,7 @@ class Inception3(nn.Module): ...@@ -147,7 +147,7 @@ class Inception3(nn.Module):
x = self.fc(x) x = self.fc(x)
# N x 1000 (num_classes) # N x 1000 (num_classes)
if self.training and self.aux_logits: if self.training and self.aux_logits:
return _InceptionOuputs(x, aux) return _InceptionOutputs(x, aux)
return x return x
......
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