pytorch
完美妖姬
这个作者很懒,什么都没留下…
展开
-
pytorch的batch normalize
torch.nn.BatchNorm1d()1. BatchNorm1d(num_features, eps = 1e-05, momentum=0.1, affine=True)对于2d或3d输入进行BN。在训练时,该层计算每次输入的均值和方差,并进行平行移动。移动平均默认的动量为0.1。在验证时,训练求得的均值/方差将用于标准化验证数据。num_features:表示输入的特征数。该期望输入的原创 2018-01-19 09:11:39 · 6829 阅读 · 0 评论 -
Torch 的几个损失函数
初步了解一下,pytorch里面的损失函数及使用方法。如有不对的地方,欢迎指出。谢谢!1. L1LossInput: X、Y可以是任意形状的输入,X与Y的 shape相同。Output: 标量用途:计算X与Y的差的绝对值, 1/n * (|Xi -Yi|)2. MSELossInput: x, y 任意具有相同shape的tensor。Output: 计算均方差3. CrossEntro...原创 2018-01-19 21:08:21 · 30176 阅读 · 3 评论