deepcrayon
/
PlantNetLibre-300K
1
0
Fork 0
Code Releases Activity

Use 12 CPUs by default

deepcrayon
Jeff Moe 2023-07-05 18:01:43 -06:00
parent d58369850e
commit 97300e76b8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

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