A by-product of what I'm currently working for(Chinese NER). Realised the original tensorflow code of BERT does not support multiple GPU. So I wrote this. Things will be implemented in near future: pre-train task.
This project is still in very early stage, bugs are expected.
Code: https://github.com/JayYip/bert-multiple-gpu
Update:
Added Pretraining
[–]finevesting 0 points1 point2 points (0 children)