• BUG
IndexError: dimension specified as 0 but tensor has no dimensions
  • 报错原因:

IndexError: dimension指定为0,但tensor没有维度
分析判定为输入模型数据维度问题

  • 评估可能出问题的点:

1- 模型本身设计问题;
2- 送入模型的训练数据维度问题;

  • 经核实模型本身无问题;
  • 找到对应的训练数据构造函数:
    果然,送入模型category_tensor为2维数据,调整(增加了一个维度)之后,问题解决:
    在这里插入图片描述
Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐