问题:
RuntimeError: CUDA error: invalid configuration argument
原因:参数过大,内存需求过大
解决方法:查看之前的代码及网络设计是否合理,是否有错误的地方。本人错误在于之前忘记了加池化层,导致数据过大,从而报错。
问题:
RuntimeError: CUDA error: invalid configuration argument
原因:参数过大,内存需求过大
解决方法:查看之前的代码及网络设计是否合理,是否有错误的地方。本人错误在于之前忘记了加池化层,导致数据过大,从而报错。