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

use cuda from argparse

parent 00bcda50
No related branches found
No related tags found
No related merge requests found
...@@ -14,7 +14,7 @@ parser.add_argument('specie', type=str, help='Target specie to detect', choices= ...@@ -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('-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('-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