From 31aa327f0e554d6950bdf4e977414107711f4f4d Mon Sep 17 00:00:00 2001 From: Nathan Fradet <56734983+Natooz@users.noreply.github.com> Date: Tue, 8 Nov 2022 16:34:00 +0100 Subject: [PATCH] update test files path --- test/test_training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_training.py b/test/test_training.py index 8a6b3a8..3ea9f32 100644 --- a/test/test_training.py +++ b/test/test_training.py @@ -80,7 +80,7 @@ def test_training(): log_cuda_info() log_model_parameters(model) - dataset = MIDIDataset(Path('test_files'), max_seq_len=128, min_seq_len=64, padding_token=0) + dataset = MIDIDataset(Path('test', 'test_files'), max_seq_len=128, min_seq_len=64, padding_token=0) subset_train, subset_valid = create_subsets(dataset, [0.4]) dataloader_train = DataLoader(subset_train, batch_size=8, collate_fn=collate_ar) dataloader_valid = DataLoader(subset_valid, batch_size=8, collate_fn=collate_ar)