摘要: 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 阅读(22) 评论(0) 推荐(0) 编辑