Add --gpu option to runtraining.sh

CRBS / cdeep3m

Please go to https://github.com/CRBS/cdeep3m2 for most recent version

Other

58 stars 10 forks source link

Closed coleslaw481 closed 5 years ago

coleslaw481 commented 5 years ago

Add --gpu # option to runtraining.sh so users can specify to use a certain GPU instead of all GPUs.

coleslaw481 commented 5 years ago

Just had to add --gpu to runtraining.sh and pass it to trainworker.sh

coleslaw481 commented 5 years ago

Fixed in 1.5.0