From 587f90214870450584b6e3947ba322040eb6da00 Mon Sep 17 00:00:00 2001
From: dengkaipeng <dengkaipeng@baidu.com>
Date: Mon, 27 Apr 2020 06:50:59 +0000
Subject: [PATCH] fix int64 error on windows

---
 examples/tsm/kinetics_dataset.py | 2 +-
 hapi/datasets/flowers.py         | 2 +-
 hapi/datasets/mnist.py           | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/examples/tsm/kinetics_dataset.py b/examples/tsm/kinetics_dataset.py
index 123d898..6cfd0d1 100644
--- a/examples/tsm/kinetics_dataset.py
+++ b/examples/tsm/kinetics_dataset.py
@@ -113,7 +113,7 @@ class KineticsDataset(Dataset):
 
         if self.transform:
             imgs, label = self.transform(imgs, label)
-        return imgs, np.array([label])
+        return imgs, np.array([label]).astype('int64')
 
     @property
     def num_classes(self):
diff --git a/hapi/datasets/flowers.py b/hapi/datasets/flowers.py
index 1f4f707..c360e8f 100644
--- a/hapi/datasets/flowers.py
+++ b/hapi/datasets/flowers.py
@@ -123,7 +123,7 @@ class Flowers(Dataset):
         if self.transform is not None:
             image, label = self.transform(image, label)
 
-        return image, label
+        return image, label.astype('int64')
 
     def __len__(self):
         return len(self.indexes)
diff --git a/hapi/datasets/mnist.py b/hapi/datasets/mnist.py
index 18c6290..11b5f31 100644
--- a/hapi/datasets/mnist.py
+++ b/hapi/datasets/mnist.py
@@ -144,7 +144,7 @@ class MNIST(Dataset):
 
                     for i in range(buffer_size):
                         self.images.append(images[i, :])
-                        self.labels.append(np.array([labels[i]]))
+                        self.labels.append(np.array([labels[i]]).astype('int64'))
 
     def __getitem__(self, idx):
         image, label = self.images[idx], self.labels[idx]
-- 
GitLab