RuntimeError: diff_view_meta->output_nr_ == 0 INTERNAL ASSERT FAILED at \“..\\\\torch\\\\csrc\\\\aut

报错:RuntimeError: diff_view_meta->output_nr_ == 0 INTERNAL ASSERT FAILED at "..\\torch\\csrc\\autograd\\variable.cpp":417, please report a bug to PyTorch.

  • 问题描述
  • 使用的训练环境
  • 问题解决
    • 一、剪枝之后出现的错误,但是自己的模型并没有剪枝的操作,故排除;
    • 二、可能是不同的版本所用语法不同造成的报错

问题描述

在pycharm中运行神经网络相关代码时,发现如下图所示的报错:
RuntimeError: diff_view_meta->output_nr_ == 0 INTERNAL ASSERT FAILED at \“..\\\\torch\\\\csrc\\\\aut_第1张图片

使用的训练环境

torch 1.8.1+cu102
torchvision 0.9.1+cu102

问题解决

在百度搜索了相关此问题的解决方案,发现有两种情况:

一、剪枝之后出现的错误,但是自己的模型并没有剪枝的操作,故排除;

二、可能是不同的版本所用语法不同造成的报错

检查报错:
RuntimeError: diff_view_meta->output_nr_ == 0 INTERNAL ASSERT FAILED at \“..\\\\torch\\\\csrc\\\\aut_第2张图片

跳转到下面所示语句后,发现多一个下划线

feat = [fm.squeeze_(1) for fm in feat]  #修改前

尝试把下划线去掉后,

feat = [fm.squeeze(1) for fm in feat]  #修改后

成功运行!
RuntimeError: diff_view_meta->output_nr_ == 0 INTERNAL ASSERT FAILED at \“..\\\\torch\\\\csrc\\\\aut_第3张图片

你可能感兴趣的:(神经网络,python,深度学习)