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

pytorch 1.0 doesn't work #21

Open
tankche1 opened this issue Feb 6, 2020 · 4 comments
Open

pytorch 1.0 doesn't work #21

tankche1 opened this issue Feb 6, 2020 · 4 comments

Comments

@tankche1
Copy link

tankche1 commented Feb 6, 2020

Good work!
Can you add a readme for pytorch1.0?
There are no layer_util in lib. so I can't bash make.sh.

@vasgaowei
Copy link
Owner

For pytorch 1.2 branch, you com compile by command: bash make_cuda.sh.

@tankche1 tankche1 closed this as completed Feb 7, 2020
@tankche1 tankche1 reopened this Feb 7, 2020
@tankche1
Copy link
Author

tankche1 commented Feb 7, 2020

I succesfully compile. Then when I train I met this, any solution?
Also check mmdetection issue and didn't find solution.
截屏2020-02-07下午4 02 21

@vasgaowei
Copy link
Owner

You can try to check if the .so file for nms_cpu exists. Also you can use nms implementated in torchvision.

@Sean-Huang65
Copy link

I met this problem too, and the .so file for nms_cpu exists....

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

3 participants