Questions about swivel implementation

25 views
Skip to first unread message

Kyle Chen

unread,
Mar 27, 2017, 4:14:39 AM3/27/17
to swivel-e...@googlegroups.com
Dear Swivel authors,

Currently, I am trying the swivel on my test problem, however I have a problem about the GPU support. It seems the current implementation on https://github.com/tensorflow/models/tree/master/swivel
requires the at least one GPU on my machine, right?

Because my test problem now is just a small one, I think the CPU is enough so far.
Will you update the source code to handle the CPU-only version?

Thanks,

Kyle

Chris Waterson

unread,
Mar 27, 2017, 2:35:20 PM3/27/17
to Kyle Chen, Swivel Embeddings
Thanks Kyle, you're right: a recent update did cause this to not work with CPU-only models. My bad!

I'll prepare a fix. You may be able to work around the problem by fat-fingering in "/cpu:0" around line 258.

--
You received this message because you are subscribed to the Google Groups "Swivel Embeddings" group.
To unsubscribe from this group and stop receiving emails from it, send an email to swivel-embeddings+unsubscribe@googlegroups.com.
To post to this group, send email to swivel-embeddings@googlegroups.com.
To view this discussion on the web visit https://groups.google.com/d/msgid/swivel-embeddings/CAAaQ1vDMHvZ_0z6aHKXg_0Nu7v6Xbzrp9jaosex8O7POWMqmbQ%40mail.gmail.com.
For more options, visit https://groups.google.com/d/optout.

Kyle Chen

unread,
Mar 27, 2017, 9:12:58 PM3/27/17
to Chris Waterson, Swivel Embeddings
Dear Chris,

Thanks for your reply,

I will work around the problem now and look forward the updates.

Best,

Kyle
Reply all
Reply to author
Forward
0 new messages