天津网站建设哪家公司好,网络营销公司模拟创建实训,做网站的时候说需求的专业术语,成都代运营公司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
