特牛网址导航

Pytorch(三):参数学习率的调整_learning rate. default=1e-4-CSDN博客

网友收藏
文章浏览阅读3.5k次,点赞3次,收藏12次。预备内容经过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, 5] 5._learning rate. default=1e-4