Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
vision
Commits
c94a1585
Commit
c94a1585
authored
Jun 17, 2019
by
taylanbil
Committed by
Soumith Chintala
Jun 17, 2019
Browse files
typo fix: ouput -> output (#1034)
I grepped the repo for Ouputs and these were the only occurences
parent
fefe118c
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
4 deletions
+4
-4
torchvision/models/googlenet.py
torchvision/models/googlenet.py
+2
-2
torchvision/models/inception.py
torchvision/models/inception.py
+2
-2
No files found.
torchvision/models/googlenet.py
View file @
c94a1585
...
@@ -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'
])
_GoogLeNetOu
t
puts
=
namedtuple
(
'GoogLeNetOu
t
puts'
,
[
'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
_GoogLeNetOu
t
puts
(
x
,
aux2
,
aux1
)
return
x
return
x
...
...
torchvision/models/inception.py
View file @
c94a1585
...
@@ -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'
])
_InceptionOu
t
puts
=
namedtuple
(
'InceptionOu
t
puts'
,
[
'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
_InceptionOu
t
puts
(
x
,
aux
)
return
x
return
x
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment