Skip to content

Commit 0955592

Browse files
David Riazatifacebook-github-bot
David Riazati
authored andcommitted
Cast nn.Upsample.scale_factor to a float (pytorch#17732)
Summary: Fixes pytorch#17106 Pull Request resolved: pytorch#17732 Differential Revision: D14388192 Pulled By: driazati fbshipit-source-id: d9c9e87a7c6db63c1de3ddebbb8dcf619f0dc34d
1 parent 4bea15f commit 0955592

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch/nn/modules/upsampling.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,7 @@ def __init__(self, size=None, scale_factor=None, mode='nearest', align_corners=N
126126
super(Upsample, self).__init__()
127127
self.name = type(self).__name__
128128
self.size = size
129-
self.scale_factor = scale_factor
129+
self.scale_factor = float(scale_factor) if scale_factor else None
130130
self.mode = mode
131131
self.align_corners = align_corners
132132

0 commit comments

Comments
 (0)