From 57892672999c2b74e19d229ab9e5dfdcc6c14b53 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Fri, 25 Oct 2024 13:33:59 +0200
Subject: [PATCH] datasets
---
main_ray_tune.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/main_ray_tune.py b/main_ray_tune.py
index 8215f01..8554204 100644
--- a/main_ray_tune.py
+++ b/main_ray_tune.py
@@ -60,7 +60,7 @@ def train_model(config, args):
data_train, data_val= common_dataset.load_data(path_train=args.dataset_train,
path_val=args.dataset_val,
path_test=args.dataset_test,
- batch_size=int(config["batch_size"]), length=25)
+ batch_size=int(config["batch_size"]), length=25, convert=True, pad = True)
else:
data_train, data_val = dataloader.load_data(data_source=args.dataset_train,
batch_size=int(config["batch_size"]), length=25)
--
GitLab