最完整的PyTorch數據科學家指南(2)

本篇文章繼續爲同學們講解神經網絡的卷積層部分。 因此,Conv2d圖層需要使用Cin通道將高度爲H且寬度爲W的圖像作爲輸入 。現在,對於卷積網絡中的第一層,的數量in_channels將爲3(RGB),並且out_channels用戶可以定義數量。kernel_size大多采用3×3是,並且stride通常使用爲1。 爲了檢查一個我不太瞭解的新層,我通常嘗試查看該層的輸入和輸出,如下所示,在該層我
相關文章
相關標籤/搜索