Change prefetch to 1 (temporary fix)

cceyda · Nov 12, 2020 · 0cc152c · 0cc152c
1 parent 342c31e
commit 0cc152c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/image_checker/cli.py b/image_checker/cli.py
@@ -34,7 +34,7 @@ def main():
     args = parser.parse_args()
     args = vars(args)
 
-    args["prefetch"]=2
+    args["prefetch"]=1
     if args["use_cpu"]:
         args["device"]="cpu"
     else: 
@@ -70,4 +70,4 @@ def main():
         raise Exception(f"Problem with logfile {e}")
 
     ds = folder_iterator(args["path"], args["extensions"], args["recursive"])
-    checker_batch(ds, args)
+    checker_batch(ds, args)