We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 40e7e26 commit bb3771bCopy full SHA for bb3771b
inception.py
@@ -110,19 +110,22 @@ def forward(self, inp):
110
Parameters
111
----------
112
inp : torch.autograd.Variable
113
- Input tensor of shape Bx3xHxW. Values are expected to be in
+ Input tensor of shape Bx3xHxW. Values are expected to be in
114
range (0, 1)
115
116
Returns
117
-------
118
- List of torch.autograd.Variable, corresponding to the selected output
+ List of torch.autograd.Variable, corresponding to the selected output
119
block, sorted ascending by index
120
"""
121
outp = []
122
x = inp
123
124
if self.resize_input:
125
- x = F.upsample(x, size=(299, 299), mode='bilinear')
+ x = F.interpolate(x,
126
+ size=(299, 299),
127
+ mode='bilinear',
128
+ align_corners=False)
129
130
if self.normalize_input:
131
x = x.clone()
0 commit comments