1. 首页
  2. 编程语言
  3. Python
  4. 用Pytorch训练CNN(数据集MNIST,使用GPU的方法)

用Pytorch训练CNN(数据集MNIST,使用GPU的方法)

上传者: 2022-04-10 03:19:37上传 PDF文件 57.89 KB 热度 17次

听说pytorch使用比TensorFlow简单,加之pytorch现已支持windows,所以今天装了pytorch玩玩,第一件事还是写了个简单的CNN在MNIST上实验,初步体验的确比TensorFlow方便。

用户评论