search

wangyuxinwhy / uniem

unified embedding model
Apache License 2.0
826 stars 64 forks source link

M3E如何指定为GPU训练，而不是CPU训练？ #128

Open sherlock1987 opened 4 months ago

sherlock1987 commented 4 months ago

🐛 bug 说明

这点比较奇怪，找了代码没找到哪里指定为gpu训练，训练的时候gpu显存有占用，但是利用率为0，这点就很奇怪。

Python Version

None

sherlock1987 commented 4 months ago

刚刚打印了一下模型的device，确实是cpu，很奇怪

EternalImmortal commented 2 months ago

+1 好像一直是cpu在训练