Do not print multi-gpu message if training on a single gpu

NVIDIA-Genomics-Research / AtacWorks

Deep learning based processing of Atac-seq data

https://clara-parabricks.github.io/AtacWorks/

Other

128 stars 23 forks source link

Do not print multi-gpu message if training on a single gpu #123

Closed avantikalal closed 4 years ago

avantikalal commented 4 years ago

When I have 8 GPUs available but I'm training on only a single GPU (with --gpu 0 and without --distributed), the message 'Distributing to 8 GPUs' is still printed.

ntadimeti commented 4 years ago

Addressed in #129