Pytroch网络参数的快速增删改查（实例讲解）

最新推荐文章于 2025-05-07 15:56:09 发布

HuiYu-Li

最新推荐文章于 2025-05-07 15:56:09 发布

阅读量362

点赞数 1

分类专栏： Pytorch

本文链接：https://blog.youkuaiyun.com/weixin_44058333/article/details/102996086

版权

Relu>Prelu

网络模型代码

class TumorNet(nn.Module):
	def __init__(self):
		super(TumorNet, self).__init__()
		self. nff = [1, 8]#NumFeature
		# forward1
		self.forward1 = nn.Sequential(
			nn.Conv3d(self.nff[0], self.nff[1], kernel_size=3, padding=1),
			nn.InstanceNorm3d(self.nff[1]),
			nn.ReLU(inplace=True),
			# nn.PReLU(),
			nn.Conv3d(self.nff[1], self.nff[1], kernel_size=3, padding=1),
			nn