Where can i configure GPU details?

arnoweng / CheXNet

A pytorch reimplementation of CheXNet

577 stars 221 forks source link

Where can i configure GPU details? #30

Closed Akhiladdh closed 5 years ago

Akhiladdh commented 5 years ago

I want to use multiple GPUs where can i configure those details?

arnoweng commented 5 years ago

@Akhiladdh The code can be directly used for training on multiple GPUs. You just need to specify available GPU IDs (e.g. _CUDA_VISIBLEDEVICES=0,1,2,3 python model.py).

Akhiladdh commented 5 years ago

@arnoweng Thank you so much. Can you add details about how to retrain this?

arnoweng commented 5 years ago

@Akhiladdh We followed the training strategy described in the official paper. Sorry the training code is not available yet.

Akhiladdh commented 5 years ago

Thank you!