From 94e067810c59b1a1b2858c2ca39fa458b8d04c90 Mon Sep 17 00:00:00 2001
From: "paul.best" <paul.best@lis-lab.fr>
Date: Thu, 13 Jul 2023 11:39:59 +0200
Subject: [PATCH] use cuda from argparse
---
run_CNN.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/run_CNN.py b/run_CNN.py
index 764bdef..ba5bfef 100644
--- a/run_CNN.py
+++ b/run_CNN.py
@@ -14,7 +14,7 @@ parser.add_argument('specie', type=str, help='Target specie to detect', choices=
parser.add_argument('-lensample', type=float, help='Length of the signal for each sample (in seconds)', default=5),
parser.add_argument('-batch_size', type=int, help='Amount of samples to process at a time (usefull for parallel computation using a GPU)', default=32),
parser.add_argument('-channel', type=int, help='Channel of the audio file to use in the model inference (starting from 0)', default=0)
-+parser.add_argument('-cuda', type=int, help='GPU id to use', default=0)
+parser.add_argument('-cuda', type=int, help='GPU id to use', default=0)
parser.add_argument('-maxPool', help='Wether to keep only the maximal prediction of each sample or the full sequence', action='store_true'),
parser.add_argument('-no-maxPool', dest='maxPool', action='store_false')
parser.add_argument('-output_filename', type=str, help='Name of the output file for saving predictions', default='')
--
GitLab