ValueError: The number of axes must be equal to the dimension of tensor.
收藏回复举报
ValueError: The number of axes must be equal to the dimension of tensor.
t('forum.solved') 已解决
发表于2023-04-13 15:12:49
0 查看

训练界面url:ModelArts-管理控制台 (huaweicloud.com)

image.png

是create_tuple_iterators用法错误还是哪里出现问题呢,传入tensor该怎么进行修改达到日志说的axes与dimension相等?

cke_4064.png

之前在关于将数据集中的图像传入做了如上操作,注释掉的部分是以前的代码,运行时报了如下的错误

cke_7928.png

界面urlModelArts-管理控制台 (huaweicloud.com)

改成

img = Tensor(np.frombuffer(pic.tobytes(), np.uint8)).astype(dtype=np.uint8)

就报了帖子里的错,请问该怎么修改。

我要发帖子