pytorch中使用Inception v3 中AttributeError: 'InceptionOutputs' object has no attribute 'dim'
时间: 2023-10-26 07:48:09 浏览: 257
pytorch之inception_v3的实现案例
这个错误通常是因为你使用的 PyTorch 版本比 Inception v3 要低。
在早期版本的 PyTorch 中,`dim()` 方法返回的是一个整数,而在后来的版本中,它返回的是一个元组。在 Inception v3 中使用了这个方法,但是在低版本的 PyTorch 中找不到这个方法,所以会出现 `AttributeError: 'InceptionOutputs' object has no attribute 'dim'` 的错误。
要解决这个问题,你可以尝试升级 PyTorch 版本到最新版本,或者修改 Inception v3 的代码,使其不再依赖于 `dim()` 方法。如果你决定修改 Inception v3 的代码,可以将 `dim()` 替换为 `size()` 方法即可。
阅读全文