From fb89cbb8fe9db07275ed9526da504c44f3de70ed Mon Sep 17 00:00:00 2001
From: nccrrv <59563903+nccrrv@users.noreply.github.com>
Date: Wed, 16 Oct 2024 16:50:35 +0800
Subject: [PATCH] Fix platform type inconsistency custom-dataset.cpp

---
 cpp/custom-dataset/custom-dataset.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/cpp/custom-dataset/custom-dataset.cpp b/cpp/custom-dataset/custom-dataset.cpp
index a9f3462f48..d85f916062 100644
--- a/cpp/custom-dataset/custom-dataset.cpp
+++ b/cpp/custom-dataset/custom-dataset.cpp
@@ -21,7 +21,7 @@ struct Options {
 
 static Options options;
 
-using Data = std::vector<std::pair<std::string, long>>;
+using Data = std::vector<std::pair<std::string, int64_t>>;
 
 class CustomDataset : public torch::data::datasets::Dataset<CustomDataset> {
   using Example = torch::data::Example<>;
@@ -73,7 +73,7 @@ std::pair<Data, Data> readInfo() {
   std::ifstream stream(options.infoFilePath);
   assert(stream.is_open());
 
-  long label;
+  int64_t label;
   std::string path, type;
 
   while (true) {