pytorch: RuntimeError: 1only batches of spatial targets supported but got targets of size : : xx

@[TOC](RuntimeError: 1only batches of spatial targets supported (non-empty 3D tensors) but got targets of size: : [64]) 问题描述 今天在使用pytorch编写修改网络结构的时候,遇到了一个问题: 如下图所示,是求网络输出和label之间交叉熵的时候遇到维度不匹配。 所以输出cla
相关文章
相关标签/搜索