Fixed cache dir for validation

This commit is contained in:
romi1502
2019-11-16 13:21:00 +01:00
parent 2bdb3aa013
commit 3ebd23649b

View File

@@ -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,