Closed coleslaw481 closed 5 years ago
Add --gpu # option to runtraining.sh so users can specify to use a certain GPU instead of all GPUs.
--gpu #
runtraining.sh
Just had to add --gpu to runtraining.sh and pass it to trainworker.sh
--gpu
trainworker.sh
Fixed in 1.5.0
Add
--gpu #
option toruntraining.sh
so users can specify to use a certain GPU instead of all GPUs.