- 博客(2)
- 资源 (2)
- 收藏
- 关注
原创 使用pytorch torch.jit.trace转换模型时出错“xxx has no attribute xxx“
在使用torch.jit.trace方法转换模型时,对于方法中的if和循环控制流是没办法处理的,这样就会出现只要产生该模块时默认都会有这个参数,但是在有些利用if来控制模型结构的地方是没有这个属性的,因此报错。解决办法:利用torch.jit.script方法转换...
2021-03-03 14:01:34 1637
原创 配置MMdetection踩过的坑:RuntimeError: nms is not compiled with GPU support
配置MMdetection踩过的坑:RuntimeError: nms is not compiled with GPU support 这个坑查阅了很多博客一般都说是pytorch版本和mmcv版本不兼容导致,因此我在配置过程中更换了很多版本的pytorch版本,仍然没有用,最终发现是mmcv下build文件夹没有删除,尤其是从同事或同学拷贝过来的代码,一定要先删除该文件夹,然后再进行环境配置。...
2021-03-01 14:11:39 5987 5
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人