-
Notifications
You must be signed in to change notification settings - Fork 174
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
No MultiLabelMarginCriterion #152
Comments
Had written this code a long time ago, just made a PR for it now: I don't have a good unit test for it though. If someone can write a quick one that would be great. |
@clementfarabet Hi, I really want to use the MultiLabelMarginCriterion on GPU. I have found your fork of cunn, however, I do not know how to use it. I mean I only know installing packages using |
hey Jireh, You can do: git clone https://github.com/clementfarabet/cunn -b master
cd cunn
luarocks make rocks/cunn-scm-1.rockspec Then you will get his version of cunn that has GPU MultiLabelMarginCriterion |
@soumith Thank you for your reply. However, I just tried exactly the same command you suggested, but failed with this error:
Do you know the possible reasons? |
I have just tried this and got the same errors above. Please advise. |
@soumith I have tried your advice and got the same errors above.any advise?or where can I find the right MultiLabelMarginCriterion.cu? |
Hi guys,
I noticed there is no cunn implementation of this criterion.
Do you plan to implement that any time soon?
We could try implementing it ourselves otherwise or just keep the criterion on CPU.
Best,
D.
The text was updated successfully, but these errors were encountered: