【深度学习入门笔记】CNN卷积神经网络 - LeNet具体实现及代码讲解
CNN卷积神经网络 - LeNet具体实现及代码讲解
本次笔记来源bilibili 霹雳吧啦Wz pytorch官方demo(Lenet)
个人github仓库:heisenbergCh/way-to-deep-learning: 个人零基础自学深度学习时产生的笔记与示例代码
参照官网demo定义LeNet网络结构
完整代码:
import torch.nn as nn
import torch.nn.functional as F
class LeNet(nn.Module):
def
原创
2022-04-25 19:34:13 ·
1906 阅读 ·
0 评论