OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Cast nn.Upsample.scale_factor to a float (#17732)

Browse Source 
Summary:
Fixes #17106
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17732

Differential Revision: D14388192

Pulled By: driazati

fbshipit-source-id: d9c9e87a7c6db63c1de3ddebbb8dcf619f0dc34d
This commit is contained in:
David Riazati 2019-03-08 15:26:25 -08:00 committed by Facebook Github Bot
parent 4bea15f580
commit 0955592243
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
torch/nn/modules/upsampling.py
View File

@ -126,7 +126,7 @@ class Upsample(Module):
super(Upsample, self).__init__() super(Upsample, self).__init__()
self.name = type(self).__name__ self.name = type(self).__name__
self.size = size self.size = size
self.scale_factor = scale_factor self.scale_factor = float(scale_factor) if scale_factor else None
self.mode = mode self.mode = mode
self.align_corners = align_corners self.align_corners = align_corners