Open natlouis opened 4 years ago
Reference to #50
Adding new config parameter: num_gpus (default: -1)
-1: Use all available GPUs 0 : Do not push model onto GPU i.e. use CPU >1: Use specified number of GPUs
Reference to #50
Adding new config parameter: num_gpus (default: -1)
-1: Use all available GPUs 0 : Do not push model onto GPU i.e. use CPU >1: Use specified number of GPUs