在学习自定义层的过程中,发现一个可能的坑
自定义层:
https://blog.csdn.net/wfei101/article/details/76735760
https://blog.csdn.net/u012426298/article/details/81016600
我在自定义的层中使用GPU果然出现错误,先不管,只用CPU,发现编译仍然不通过。
该问题通过,将之前同时导入的.cu文件删除来解决。
在学习自定义层的过程中,发现一个可能的坑
自定义层:
https://blog.csdn.net/wfei101/article/details/76735760
https://blog.csdn.net/u012426298/article/details/81016600
我在自定义的层中使用GPU果然出现错误,先不管,只用CPU,发现编译仍然不通过。
该问题通过,将之前同时导入的.cu文件删除来解决。