Bug日常
一天一bug,早日升级
西檬饭
这个作者很懒,什么都没留下…
展开
-
使用单机单机多卡的训练模型时复信号输入问题
项目场景:使用单机多卡的模式训练模型问题描述torch.complex类型数据为模型输入,单卡训练正常,多卡训练报错。原因分析:单机多卡训练过程:解决方案:提示:这里填写该问题的具体解决方案:例如:新建一个 Message 对象,并将读取到的数据存入 Message,然后 mHandler.obtainMessage(READ_DATA, bytes, -1, buffer).sendToTarget();换成 mHandler.sendMessage()。...原创 2022-03-20 15:42:16 · 367 阅读 · 0 评论 -
AttributeError: module ‘torch.onnx‘ has no attribute ‘set_training‘
问题描述import torchimport tensorwatch as twresnet18 = models.resnet18(pretrained=True)tw.draw_model(resnet18, (1, 3, 224, 224))使用tensorwatch展示网络结构时报错。问题原因当前环境中Pytorch版本为1.8。高于1.6的版本中,模块’torch.onnx’去掉了 'set_training’这个属性。解决办法将Pytorch降到1.6以下。Conda 安原创 2021-06-23 10:23:13 · 3066 阅读 · 7 评论 -
ValueError: Shape mismatch: The shape of labels (received (320,)) should equal the shape of logits e
问题描述ValueError: Shape mismatch: The shape of labels (received (320,)) should equal the shape of logits except for the last dimension (received (64, 5)).产生原因这个问题是由于输出层的类别数和训练数据shape不同导致。在训练模型时,一般会将...原创 2020-03-20 00:16:30 · 2867 阅读 · 0 评论 -
使用lstm中的stateful和return_sequence导致InvalidArgumentError
问题描述训练一个模型,使用LSTM组为第一层,具体模型如下图。当使用LSTM的默认参数为 stateful = False,return_sequences = False时,能够正常训练。然而当要返回每步的状态时,即LSTM的默认参数为 stateful = True,return_sequences = True时,模型和报错如下:InvalidArgumentError ...原创 2020-03-11 15:25:42 · 857 阅读 · 1 评论 -
tensorflow.python.framework.errors_impl.InvalidArgumentError
尝试使用一个for循环训练10次神经网络取10次结果作为均值,for循环第一次能够正常运行,当i =1时,报错:tensorflow.python.framework.errors_impl.InvalidArgumentError:You must feed a value for placeholder tensor ‘Placeholder’ with dtype float[[nod...原创 2019-02-11 20:36:30 · 5284 阅读 · 0 评论 -
Tensorflow TFRecord: Can't parse serialized example
问题描述InvalidArgumentError: Key: 'label’. Can’t parse serialized Example.保存tfrecorddef save_tfrecords(data, label, desfile): assert data.shape[0] == label.shape[0] with tf.python_io.TFRecordW...原创 2019-02-16 22:48:07 · 8483 阅读 · 0 评论 -
tf.train.Feature() TypeError
问题描述:将npy文件存为tfrecord文件时候,保存label的时候报错:“label”: tf.train.Feature(int64_list=tf.train.Int64List(value=[label[i]]))TypeError: only size-1 arrays can be converted to Python scalars产生原因:def save_tfre...原创 2019-02-16 22:28:00 · 4917 阅读 · 0 评论 -
Anaconda找包, PackageNotFoundError: ''Package missing in current channels"
Anaconda作为一个工具包集成管理工具,下载python工具包是很方便的,直接敲:conda install package_name但是有时候安装一个工具包(如skimage)的时候,在当前的channels中找不到这个包,会提示:PackageNotFoundError: Packages missing in current channels: - skimageWe ...转载 2019-02-02 02:17:43 · 1146 阅读 · 1 评论 -
AttributeError: module 'pip' has no attribute 'main'
问题描述: 安装第三方包的时候遇到的 AttributeError: module ‘pip’ has no attribute ‘main’ 找到安装目录下 helpers/packaging_tool.py文件,找到如下代码:def do_install(pkgs): try: import pip except ImportError: ...转载 2018-06-13 15:28:38 · 486 阅读 · 0 评论 -
Window下安装 hmmlearn 问题: Microsoft visual C++ 14.0 is required
(1) 问题描述: 安装Python包hmmlearn失败,遇到error: Microsoft visual C++ 14.0 is required. (2) 解决方法: 我查看了安装的软件, 难道这个不是要的?: 于是重新装了Microsoft Visual C++ 2017. 选择了通用安装。 查看control pannel/programs ...原创 2018-05-23 17:28:15 · 1173 阅读 · 0 评论