錯(cuò)誤

Pytorch RuntimeERROR: Given groups=1 weights of size [256,64,1,1] expected input[1,16,256,256] to have 64 channels, but got 16 channels instead.

錯(cuò)誤分析

Given groups=1 weights of size [256,64,1,1]

代表卷積核的channel 大小為 64->256 ，大小為1*1

expected input [1,16,256,256] to have 64 channels

代表現(xiàn)在要卷積的feature的大小，channel為16, 其實(shí)我們期望的輸入feature大小channel 為64個(gè)通道

but got 16 channel instead

代表我們得到16個(gè)channels 的feature 與預(yù)期64個(gè)channel不一樣。

總結(jié)

以上是生活随笔為你收集整理的Pytorch RuntimeERROR: Given groups=1 weights of size [256,64,1,1] expected input[1,16,256,256] to的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。