From 51880db93c5942a0715d2c5fa8dd46896b42534a Mon Sep 17 00:00:00 2001
From: Yandi <yandirzm@gmail.com>
Date: Sun, 22 Jan 2023 17:31:26 +0100
Subject: [PATCH] [Dev] Fixed bug on test loader
---
dataset.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/dataset.py b/dataset.py
index c814c97..63c9f51 100644
--- a/dataset.py
+++ b/dataset.py
@@ -326,7 +326,7 @@ class Dataset(data.Dataset):
return in_values, out_values
else:
- in_values = torch.Tensor(values[:, :-1])
+ in_values = torch.Tensor(values)
if self.transform is not None:
in_values = self.transform(in_values)
return in_values
--
GitLab