zoukankan      html  css  js  c++  java
  • pytorch踩坑记

    因为我有数学物理背景,所以清楚卷积的原理。但是在看pytorch文档的时候感到非常头大,罗列的公式以及各种令人眩晕的下标让入门新手不知所云...最初我以为torch.nn.conv1d的参数in_channel/out_channel表示图像的通道数,经过运行错误提示之后,才知道[in_channel,kernel_size]构成了卷积核。 

    loss函数中target/label的类型必须是dtype=torch.long 

    在使用cross_entropy时出现了一个错误RuntimeError: Dimension out of range (expected to be in range of [-1, 0], but got 1),查看functional.py的1550行中:
    其中log_softmax(input, 1)指定了dim=1,而log_softmax实现如下:

    排查之后发现,cross_entropy输入的矩阵行数需要等于taget/label的个数,即行对行。

  • 相关阅读:
    0x05 排序
    bzoj3032: 七夕祭
    0x04 二分
    bzoj2783: [JLOI2012]树
    bzoj3192: [JLOI2013]删除物品
    bzj1106: [POI2007]立方体大作战tet
    POJ2299Ultra-QuickSort
    POJ3080Blue Jeans
    POJ3253Babelfish
    POJ1611The Suspects
  • 原文地址:https://www.cnblogs.com/webbery/p/10454173.html
Copyright © 2011-2022 走看看