2023年7月11日
pytorch保存模型及加载模型
摘要： ```python Class TestModle(nn.Module): def __init__(self): self.conv = nn.Conv(3, 6, 5) self.pool = nn.MaxPool2d(2, 2) ... def forward(self, x): ... .. 阅读全文
posted @ 2023-07-11 17:37 waterdoor 阅读(29) 评论(0) 推荐(0) 编辑