From 254bc38e61445f4664021e165130c6a6db60dbd9 Mon Sep 17 00:00:00 2001
From: schne <leo.schneider@ecl19.ec-lyon.fr>
Date: Mon, 9 Sep 2024 12:40:08 +0200
Subject: [PATCH] dataset
---
dataloader.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/dataloader.py b/dataloader.py
index d3b872e..121ad91 100644
--- a/dataloader.py
+++ b/dataloader.py
@@ -77,8 +77,7 @@ class RT_Dataset(Dataset):
self.data = self.data[self.data.state == 'holdout']
elif mode == 'validation':
self.data = self.data[self.data.state == 'validation']
- if size is not None:
- self.data = self.data.sample(size)
+
print('Padding')
self.data['sequence'] = self.data['sequence'].str.pad(length, side='right', fillchar='_')
--
GitLab