- 博客(1)
- 收藏
- 关注
RSS订阅原创 pytorch 保存与读取模型块
class f(nn.Module): def __init__(self): super(f, self).__init__() self.con0 = nn.Sequential( nn.Conv2d(3, 96, 1), nn.LeakyReLU(0.03, inplace=True), nn.Conv2d(96, 96, 3) ) 实例化： F = f() 保存： to
2021-04-17 14:21:54
60
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人