MichiganCOG / ViP

Video Platform for Action Recognition and Object Detection in Pytorch
MIT License
220 stars 37 forks source link

Support for multi-gpu or cpu-only training #51

Open natlouis opened 4 years ago

natlouis commented 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