随笔分类 -  pytorch

Pytorch_LeNet
摘要：1 model定义 import torch.nn as nn import torch.nn.functional as F # 定义类，继承nn.Module父类，实现init函数和forward函数（正向传播） class LeNet(nn.Module): def __init__(self 阅读全文
posted @ 2022-08-26 16:42 Truman001 阅读(26) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示