摘要: 环境: pytorch1.1 cuda9.0 ubuntu16.04 该网络有3层,第一层input layer,有784个神经元(MNIST数据集是28*28的单通道图片,故有784个神经元)。第二层为hidden_layer,设置为500个神经元。最后一层是输出层,有10个神经元(10分类任务) 阅读全文
posted @ 2019-08-14 22:47 liualex_sone 阅读(5217) 评论(3) 推荐(0) 编辑