Pytorch Note8 简单介绍torch.optim(优化)和模型保存
全部笔记的汇总贴:Pytorch Note 快乐星球
nn.Module(模组)
在PyTorch里面编写神经网络,所有的层结构和损失函数都来自于torch.nn,所有的模型构建都是从这个基类nn.Module继承的,于是有了下面这个模板。
import torch.nn as nn
class net_name(nn.Module)
全部笔记的汇总贴:Pytorch Note 快乐星球
在PyTorch里面编写神经网络,所有的层结构和损失函数都来自于torch.nn,所有的模型构建都是从这个基类nn.Module继承的,于是有了下面这个模板。
import torch.nn as nn
class net_name(nn.Module)