PyTorch 分类实现(MNIST)——read data from Image
0. 写作目的
好记性不如烂笔头。
1. 直接上代码
#!/usr/bin/env python
# _*_ coding:utf-8 _*_
import torch
import torch.nn as nn ## 各层类型的实现
import torch.nn.functional as F ## 各层函数的实现,与层类型对应, 如:卷积函数/池化函数/归一化函数等
from...
原创
2018-08-21 17:46:16 ·
728 阅读 ·
0 评论