PyTorch定义模型中用到的self.register_buffer()

PyTorch中定义模型时,有时会遇到self.register_buffer('name',Tensor)的操作,该方法的作用是定义一组参数,该组参数的特别之处在于:模型训练时不会更新,即调用optimizer.step()后改组参数不会发生变化,只可人为地改变,但在保存模型时,该组参数又作为模型参数不可或缺的一部分被保存。

你可能感兴趣的:(pytorch,深度学习,人工智能)