From 2e440e0f9c687314d5ff67ead68ce985dc446e3a Mon Sep 17 00:00:00 2001 From: Zhiqing Sun Date: Fri, 26 Jun 2020 09:49:39 -0400 Subject: [PATCH] Update dataloader.py --- codes/dataloader.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/codes/dataloader.py b/codes/dataloader.py index 70d43a25..ed3f3492 100644 --- a/codes/dataloader.py +++ b/codes/dataloader.py @@ -59,8 +59,8 @@ def __getitem__(self, idx): negative_sample = np.concatenate(negative_sample_list)[:self.negative_sample_size] - negative_sample = torch.from_numpy(negative_sample) - + negative_sample = torch.LongTensor(negative_sample) + positive_sample = torch.LongTensor(positive_sample) return positive_sample, negative_sample, subsampling_weight, self.mode @@ -181,4 +181,4 @@ def one_shot_iterator(dataloader): ''' while True: for data in dataloader: - yield data \ No newline at end of file + yield data