Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

basset_motifs_predict.lua does not work with GPU-trained model #6

Closed
jmrinaldi opened this issue Dec 2, 2015 · 1 comment
Closed

Comments

@jmrinaldi
Copy link

basset_motifs_predict.lua expects a CPU-trained model, making basset_motifs.py non-functional for GPU-trained models.

Converting all model tensors to double solves problem, may want to add this functionality

@davek44
Copy link
Owner

davek44 commented Jan 19, 2016

Ah great. In general, moving models back and forth between the GPU and CPU can be tricky, so I don't want to make a ton of promises there. I added support for a GPU model to that script though.

@davek44 davek44 closed this as completed Jan 19, 2016
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants