search

namisan / mt-dnn

Multi-Task Deep Neural Networks for Natural Language Understanding
MIT License
2.22k stars 411 forks source link

question about multi-gpu training #174

Closed Akeepers closed 4 years ago

Akeepers commented 4 years ago

I doubt that whether mt-dnn can work normally with multi-GPU？

For example (2 gpu), at the same time:

Does it work?

namisan commented 4 years ago

For multi GPU, use this flag: --multi_gpu_on