代码小结(4.11)

1.torch.nn.Variable()

  其中不仅包含了data,还包含了grad,grad_fn.

关于Variable

2.model.train()和model.eval()

model.train()训练时启用Batch Normalization和Dropout。

model.eval()在验证集和测试集上不启用Batch Normalization和Dropout。

model.train()和model.eval()

3.torch.optim.Adam()

torch.optim.Adam方法参数

4.torch.nn.LSTM()和torch.nn.GRU()

LSTM和GRU

5.torch.nn.Parameter()

  将Tensor类型转化为可训练的参数类型Parameter,并将这个Parameter类型绑定到model上。

torch.nn.Parameter()

6.torch.nn.Conv2d()

  卷积核个数决定了输出的深度;卷积层的前一层,有多少个特征图,卷积核的通道就有多少个。

torch.nn.Conv2d()

7.torch.nn.ModuleList()

  类似于一个list类型,可以添加和移除元素,只不过其中的元素是模型的子类。

torch.nn.ModuleList()

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值