Hierarchical-Multi-Label-Text-Classification icon indicating copy to clipboard operation
Hierarchical-Multi-Label-Text-Classification copied to clipboard
verified publisher icon RandolphVI

使用GPU训练需要修改哪些参数

Open Tesla-jiang opened this issue 4 years ago • 0 comments

感谢您的工作 我已经在电脑上成功运行了代码，但是貌似只在CPU上工作，我已经设置了gpu-options-allow-growth=True，且能看到有一部显存占用，但似乎训练都是再CPU上完成的，我需要调整别的什么配置么?多谢

Tesla-jiang avatar Dec 16 '21 07:12 Tesla-jiang