From: David Riazati <davidriazati@fb.com>
Date: Fri, 8 Mar 2019 23:26:25 +0000 (-0800)
Subject: Cast nn.Upsample.scale_factor to a float (#17732)
X-Git-Tag: accepted/tizen/6.5/unified/20211028.231830~914
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=0955592243b8ed122a14ab6c1cd8e02ed49ec6d9;p=platform%2Fupstream%2Fpytorch.git

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

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

Differential Revision: D14388192

Pulled By: driazati

fbshipit-source-id: d9c9e87a7c6db63c1de3ddebbb8dcf619f0dc34d
---

diff --git a/torch/nn/modules/upsampling.py b/torch/nn/modules/upsampling.py
index da7b57d..38f98c0 100644
--- a/torch/nn/modules/upsampling.py
+++ b/torch/nn/modules/upsampling.py
@@ -126,7 +126,7 @@ class Upsample(Module):
         super(Upsample, self).__init__()
         self.name = type(self).__name__
         self.size = size
-        self.scale_factor = scale_factor
+        self.scale_factor = float(scale_factor) if scale_factor else None
         self.mode = mode
         self.align_corners = align_corners