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(