建站seo推广,研发网站建设报价,wordpress flatsome,西安黄页网分类目录#xff1a;《深入浅出Pytorch函数》总目录 torch.nn.Module是所有Pytorch中所有神经网络模型的基类#xff0c;我们的神经网络模型也应该继承这个类。Modules可以包含其它Modules#xff0c;也允许使用树结构嵌入他们#xff0c;还可以将子模块赋值给模型属性。
…分类目录《深入浅出Pytorch函数》总目录 torch.nn.Module是所有Pytorch中所有神经网络模型的基类我们的神经网络模型也应该继承这个类。Modules可以包含其它Modules也允许使用树结构嵌入他们还可以将子模块赋值给模型属性。
实例
import torch.nn as nn
import torch.nn.functional as Fclass Model(nn.Module):def __init__(self):super(Model, self).__init__()self.conv1 nn.Conv2d(1, 20, 5) # submodule: Conv2dself.conv2 nn.Conv2d(20, 20, 5)def forward(self, x):x F.relu(self.conv1(x))return F.relu(self.conv2(x))通过上面方式赋值的submodule会被注册。当调用.cuda()的时候submodule的参数也会转换为cuda Tensor。
函数
eval()
将模块设置为evaluation模式相当于self.train(False)。这个函数仅当模型中有Dropout和BatchNorm时才会有影响。
def eval(self: T) - T:rSets the module in evaluation mode.This has any effect only on certain modules. See documentations ofparticular modules for details of their behaviors in training/evaluationmode, if they are affected, e.g. :class:Dropout, :class:BatchNorm,etc.This is equivalent with :meth:self.train(False) torch.nn.Module.train.See :ref:locally-disable-grad-doc for a comparison between.eval() and several similar mechanisms that may be confused with it.Returns:Module: selfreturn self.train(False)
