File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff line change @@ -110,19 +110,22 @@ def forward(self, inp):
110
110
Parameters
111
111
----------
112
112
inp : torch.autograd.Variable
113
- Input tensor of shape Bx3xHxW. Values are expected to be in
113
+ Input tensor of shape Bx3xHxW. Values are expected to be in
114
114
range (0, 1)
115
115
116
116
Returns
117
117
-------
118
- List of torch.autograd.Variable, corresponding to the selected output
118
+ List of torch.autograd.Variable, corresponding to the selected output
119
119
block, sorted ascending by index
120
120
"""
121
121
outp = []
122
122
x = inp
123
123
124
124
if self .resize_input :
125
- x = F .upsample (x , size = (299 , 299 ), mode = 'bilinear' )
125
+ x = F .interpolate (x ,
126
+ size = (299 , 299 ),
127
+ mode = 'bilinear' ,
128
+ align_corners = False )
126
129
127
130
if self .normalize_input :
128
131
x = x .clone ()
You can’t perform that action at this time.
0 commit comments