ViP icon indicating copy to clipboard operation
ViP copied to clipboard

Support for multi-gpu or cpu-only training

Open natlouis opened this issue 6 years ago • 0 comments

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

  • Automatically loads weights regardless of whether it came from single or multi-gpu training.
  • Config batch size is multiplied by the number of GPUs

natlouis avatar Dec 31 '19 21:12 natlouis