Skip to content

Commit 22d65a4

Browse files
committed
refactor: Apply linting
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent 76db42c commit 22d65a4

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/accuracy/datasets/cifar10.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,8 @@ std::pair<torch::Tensor, torch::Tensor> read_batch(const std::string& path) {
5151
labels.push_back(label);
5252
auto image_tensor =
5353
torch::from_blob(image.data(), {kImageChannels, kImageDim, kImageDim}, torch::TensorOptions().dtype(torch::kU8))
54-
.to(torch::kF32).div(255);
54+
.to(torch::kF32)
55+
.div(255);
5556
images.push_back(image_tensor);
5657
}
5758

0 commit comments

Comments
 (0)