updated njobs

biomap/helper.py

@@ -51,7 +51,9 @@ def inference_on_location(model, longitude=2.98, latitude=48.81, start_date=2020
     
     dates = [d.strftime("%Y-%m-%d") for d in dates]
 
-    all_image = Parallel(n_jobs=cpu_count(), prefer="threads")(delayed(get_image)(location, d1,d2) for d1, d2 in zip(dates[:-1],dates[1:]))
+    n_jobs = min([12, len(dates)])
+
+    all_image = Parallel(n_jobs=n_jobs, prefer="threads")(delayed(get_image)(location, d1,d2) for d1, d2 in zip(dates[:-1],dates[1:]))
     # all_image = [cv2.imread("output/img.png") for i in range(len(dates))]
     outputs = inference(np.array(all_image), model)