#16627 convert weights using torch.as_tensor to avoid warning (#17067)

Summary:
Minor change which fixes #16627
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17067

Differential Revision: D14078726

Pulled By: soumith

fbshipit-source-id: c04a5f1eff44e4a4b04b981f0ae8de6ff018515b

diff --git a/torch/utils/data/sampler.py b/torch/utils/data/sampler.py
index d0ecdb4..ea0564c 100644 (file)
--- a/torch/utils/data/sampler.py
+++ b/torch/utils/data/sampler.py
@@ -116,7 +116,7 @@ class WeightedRandomSampler(Sampler):
         if not isinstance(replacement, bool):
             raise ValueError("replacement should be a boolean value, but got "
                              "replacement={}".format(replacement))
-        self.weights = torch.tensor(weights, dtype=torch.double)
+        self.weights = torch.as_tensor(weights, dtype=torch.double)
         self.num_samples = num_samples
         self.replacement = replacement