Skip to content

Comments

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

Open
natlouis wants to merge 6 commits intodevfrom
multi_gpu
Open

Support for multi-gpu or cpu-only training#51
natlouis wants to merge 6 commits intodevfrom
multi_gpu

Conversation

@natlouis
Copy link
Collaborator

@natlouis natlouis commented Dec 31, 2019

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 natlouis requested a review from zeonzir December 31, 2019 21:43
@natlouis natlouis changed the base branch from master to dev December 31, 2019 21:43
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant