refactor: Add dataset scaling

narendasan · narendasan · commit 15f9205369d1 · 2021-08-12T12:14:45.000-07:00
Signed-off-by: Naren Dasan &lt;naren@narendasan.com&gt;
Signed-off-by: Naren Dasan &lt;narens@nvidia.com&gt;
diff --git a/tests/accuracy/datasets/cifar10.cpp b/tests/accuracy/datasets/cifar10.cpp
@@ -51,7 +51,7 @@ std::pair<torch::Tensor, torch::Tensor> read_batch(const std::string& path) {
     labels.push_back(label);
     auto image_tensor =
         torch::from_blob(image.data(), {kImageChannels, kImageDim, kImageDim}, torch::TensorOptions().dtype(torch::kU8))
-            .to(torch::kF32);
+            .to(torch::kF32).div(255);
     images.push_back(image_tensor);
   }
 

Original file line number	Diff line number	Diff line change
`@@ -51,7 +51,7 @@ std::pair<torch::Tensor, torch::Tensor> read_batch(const std::string& path) {`
`51`	`51`	`labels.push_back(label);`
`52`	`52`	`auto image_tensor =`
`53`	`53`	`torch::from_blob(image.data(), {kImageChannels, kImageDim, kImageDim}, torch::TensorOptions().dtype(torch::kU8))`
`54`		`- .to(torch::kF32);`
	`54`	`+ .to(torch::kF32).div(255);`
`55`	`55`	`images.push_back(image_tensor);`
`56`	`56`	`}`
`57`	`57`