Pytorch(三):參數學習率的調整

預備內容 經過pytorch(一):torch構建數據集並訓練一個神經網絡,構建了一個神經網絡。其中網絡的架構如下: Layer Layer (type) Output Shape 0 (input) [2, 1, 8, 8] 1 Conv2d [2, 4, 9, 9] 2 ReLU [2, 4, 9, 9] 3 MaxPool2d [4, 4, 4, 4] 4 Conv2d [4, 8, 5,
相關文章
相關標籤/搜索