From 97300e76b8d227acf6f57ec6ffafa16dc5267fc6 Mon Sep 17 00:00:00 2001
From: Jeff Moe <moe@spacecruft.org>
Date: Wed, 5 Jul 2023 18:01:43 -0600
Subject: [PATCH] Use 12 CPUs by default

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index d8e4fb6..cfb91db 100755
--- a/train.py
+++ b/train.py
@@ -54,7 +54,7 @@ ROOT_DIR = "/srv/ml/plantnet/files/plantnet_300K/images"
 # Use GPU
 USE_GPU = "1"
 # Use all CPUs available on system. XXX get nproc
-NUM_WORKERS = "4"
+NUM_WORKERS = "12"
 
 # Parse command line options
 parser = argparse.ArgumentParser(