天津网站建设哪家公司好,成都市住房与城乡建设厅网站,国内大型网站域名,自己的域名搭建网站PyTorch中定义模型时#xff0c;有时候会遇到self.register_buffer(name, Tensor)的操作#xff0c;该方法的作用是定义一组参数#xff0c;该组参数的特别之处在于#xff1a;模型训练时不会更新#xff08;即调用 optimizer.step() 后该组参数不会变化#xff0c;只可人…PyTorch中定义模型时有时候会遇到self.register_buffer(name, Tensor)的操作该方法的作用是定义一组参数该组参数的特别之处在于模型训练时不会更新即调用 optimizer.step() 后该组参数不会变化只可人为地改变它们的值但是保存模型时该组参数又作为模型参数不可或缺的一部分被保存。
但是在使用时发现buffer里的值发生变化了。代码如下 grid_x self.Thetagrid_x[:,:,:,0] grid_x[:,:,:,0] / (self.retinal_W/2) - 1
解决办法为使用clone grid_x self.Theta.clone()grid_x[:,:,:,0] grid_x[:,:,:,0] / (self.retinal_W/2) - 1
