From c60e0861ed09a1f1215614d8b2696faacf25a756 Mon Sep 17 00:00:00 2001
From: Colin Brisson <colibrisson@gmail.com>
Date: Wed, 31 May 2023 07:52:57 +0000
Subject: [PATCH] Fix ketos test

---
 kraken/ketos/recognition.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/kraken/ketos/recognition.py b/kraken/ketos/recognition.py
index eeca4ee2e..9ece029e8 100644
--- a/kraken/ketos/recognition.py
+++ b/kraken/ketos/recognition.py
@@ -464,7 +464,7 @@ def test(ctx, batch_size, model, evaluation_files, device, pad, workers,
         ds_loader = DataLoader(ds,
                                batch_size=batch_size,
                                num_workers=workers,
-                               pin_memory=True,
+                               pin_memory=False,
                                collate_fn=collate_sequences)
 
         with KrakenProgressBar() as progress: