torch如何查看模型的中間輸出結果

         博主之前一直是用tensorflow,但是最近有在接觸一點torch,發現torch中有一個地方很不方便就是他定義在網絡中間的層的結果我們沒法看,因爲它沒有辦法像tensorflow這種可以直接通過名字索引。比如: self.conv1_1 = self.conv_layer(bgr, 3, 64, "conv1_1") self.conv1_2 = self.conv_laye
相關文章
相關標籤/搜索