search

Tencent / NeuralNLP-NeuralClassifier

An Open-source Neural Hierarchical Multi-label Text Classification Toolkit
Other
1.83k stars 402 forks source link

visible_device_list #24

Closed kianaakbari closed 4 years ago

kianaakbari commented 4 years ago

Hi, in case of having multiple GPUs, how should we specify them as visible_device_list in config file? For example for GPUs with indexes 0 to 2, should we place them in a list like [0,1,2] or some special delimiter/format must be used? Thank you

liqunhit commented 4 years ago

format like："0,1,2"

coderbyr commented 4 years ago

Actually, this toolkit only support single GPU, we'll consider support running on multiple GPUs in future.