经过“ Pytorch 之 torch.nn初探“ 实训的学习,想必同学们对torch.nn
有了一个初步的认识。接下来,本实训将介绍更多内容帮助同学们运用神经网络的特性。
任务描述
本关任务:
本关提供了一个Variable
类型的变量input
,要求利用BatchNorm1d
创建一个4
维的 带有学习参数的正则化量 m
,并输出其weight
和bias
。
相关知识
正则化项即罚函数,该项对模型向量进行“惩罚”,从而避免单纯最小二乘问题的过拟合问题。在本节中,我们将介绍正则化方面的知识,这有助于预防过拟合的问题的产生。
首先,先引入我们要使用的包
import torch
import torch.nn
from torch.autograd import Variable
BatchNorm
BatchNorm | 描述 |
---|---|
BatchNorm1d | 将2维或3维的输入视为mini-batch,应用 Batch 正则化 |
BatchNorm2d | 将4维的输入视为3维的mini-batch,应用 Batch 正则化 |
BatchNorm3d | 将5维的输入视为4维的mini-batch,应用 Batch 正则化 |
下面以 |