Skip to content
Snippets Groups Projects
Commit 00bcda50 authored by Paul Best's avatar Paul Best
Browse files

use cuda from argparse

parent 3aafb6ac
Branches
No related tags found
No related merge requests found
...@@ -14,6 +14,7 @@ parser.add_argument('specie', type=str, help='Target specie to detect', choices= ...@@ -14,6 +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('-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('-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('-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('-maxPool', help='Wether to keep only the maximal prediction of each sample or the full sequence', action='store_true'), 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('-no-maxPool', dest='maxPool', action='store_false')
parser.add_argument('-output_filename', type=str, help='Name of the output file for saving predictions', default='') parser.add_argument('-output_filename', type=str, help='Name of the output file for saving predictions', default='')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment