From 3ebd23649b79877505ef0e9aca8661cd29ce9451 Mon Sep 17 00:00:00 2001 From: romi1502 Date: Sat, 16 Nov 2019 13:21:00 +0100 Subject: [PATCH] Fixed cache dir for validation --- spleeter/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spleeter/dataset.py b/spleeter/dataset.py index dc65652..236923a 100644 --- a/spleeter/dataset.py +++ b/spleeter/dataset.py @@ -102,7 +102,7 @@ def get_validation_dataset(audio_params, audio_adapter, audio_path): return builder.build( audio_params.get('validation_csv'), batch_size=audio_params.get('batch_size'), - cache_directory=audio_params.get('training_cache'), + cache_directory=audio_params.get('validation_cache'), convert_to_uint=True, infinite_generator=False, n_chunks_per_song=1,