深度学习:el tamaño de la vista no es compatible con el tamaño y la zancada del tensor de entrada (al menos una dimensión abarca un

Mensaje de error:

view size is not compatible with input tensor's size and stride (at least one dimension spans across two contiguous subspaces). Use .reshape(...) instead.

código de error:

prec1 = accuracy(-1 * dist_xt_ct.data, target_targets, topk=(1,))[0].item()

Código correcto:

prec1 = accuracy(-1 * dist_xt_ct.data, target_targets, topk=(1,)).item()

¡Simplemente elimine [0]!

Supongo que te gusta

Origin blog.csdn.net/weixin_44575717/article/details/124119167
Recomendado
Clasificación