关于LibTorch训练参数无法更新

最新推荐文章于 2024-05-19 01:06:15 发布

言和真

最新推荐文章于 2024-05-19 01:06:15 发布

阅读量263

点赞数

分类专栏：编程记录文章标签： PyTorch 模型注册构造函数 NormModuleImpl Adam优化器

本文链接：https://blog.youkuaiyun.com/qq_39969166/article/details/113026236

版权

编程记录专栏收录该内容

10 篇文章

订阅专栏

可能原因是
构造函数里没有注册模型到参数中

NormModuleImpl::NormModuleImpl() :
	mc{
	 torch::nn::Conv2d(torch::nn::Conv2dOptions(1,32,3).stride(1).padding(1)),
			
	}
{	
	//没写这句代码 模型参数不work
	register_module("ds", mc);
}

int main(){

	Adam optimizer(ae->parameters());
	return 0;
}