diff --git a/main_ray_tune.py b/main_ray_tune.py index 8215f01284d48b70023dadfdf9dc3023719d111c..85542043af7240001774ed7e684db8b718242c85 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)