admin 管理员组

文章数量: 887042

pytorch错误记录：TypeError: Cannot handle this data type: (1, 1, 64),

经过测试：

pairwise_distance函数在pytorch1.7上的表现与pytorch1.10/pytorch1.11上有区别,代码如下

pytorch1.7上

feat_A的size是1，64，64，64，feat_B的size是1，64，64，64

出来应该size是1，1，64，64

而在pytorch1.10/pytorch1.11

出来的size是1，64，64，1

import torch.nn.functional as Fself.dist = F.pairwise_distance(self.feat_A, self.feat_B, keepdim=True)

直接导致了进一步的计算图像大小存不出来

image_pil = Image.fromarray(image_numpy)File "D:\software\Anaconda3\envs\pt3.8\lib\site-packages\PIL\Image.py", line 2815, in fromarrayraise TypeError("Cannot handle this data type: %s, %s" % typekey) from e
TypeError: Cannot handle this data type: (1, 1, 64), |u1

解决方式只能是版本回退

本文标签： pytorch错误记录TypeError Cannot handle this data type (1 1 64)

版权声明：本文标题：pytorch错误记录：TypeError: Cannot handle this data type: (1, 1, 64), 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.freenas.com.cn/free/1696796996h257957.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。