本文整理匯總了Python中torch.nn.Sequentail方法的典型用法代碼示例。如果您正苦於以下問題：Python nn.Sequentail方法的具體用法？Python nn.Sequentail怎麽用？Python nn.Sequentail使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類torch.nn的用法示例。
在下文中一共展示了nn.Sequentail方法的2個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: add_stage
# 需要導入模塊: from torch import nn [as 別名]
# 或者: from torch.nn import Sequentail [as 別名]
def add_stage(inplanes, outplanes, innerplanes, nblocks, dilation=1, stride_init=2):
"""Make a stage consist of `nblocks` residual blocks.
Returns:
- stage module: an nn.Sequentail module of residual blocks
- final output dimension
"""
res_blocks = []
stride = stride_init
for _ in range(nblocks):
res_blocks.append(add_residual_block(
inplanes, outplanes, innerplanes, dilation, stride
))
inplanes = outplanes
stride = 1
return nn.Sequential(*res_blocks), outplanes 示例2: add_stage
# 需要導入模塊: from torch import nn [as 別名]
# 或者: from torch.nn import Sequentail [as 別名]
def add_stage(inplanes, outplanes, innerplanes, nblocks, dilation=1, stride_init=2, deform=False, nres=0):
"""Make a stage consist of `nblocks` residual blocks.
Returns:
- stage module: an nn.Sequentail module of residual blocks
- final output dimension
"""
res_blocks = []
stride = stride_init
for i in range(nblocks):
if (nres == 3 or nres == 4) and i == (nblocks-1):
res_blocks.append(add_residual_block(
inplanes, outplanes, innerplanes, dilation, stride, deform=deform
))
else:
res_blocks.append(add_residual_block(
inplanes, outplanes, innerplanes, dilation, stride, deform=deform
))
inplanes = outplanes
stride = 1
return nn.Sequential(*res_blocks), outplanes