Pytorch模型中的parameter与buffer【转】

原文:Pytorch模型中的parameter与buffer

详细的参看原文,解释的相当的nice!评论也很OK的~

说白了`parameter`注册的变量可以被optimizer.step更新(前提是requirements_grad=True)

register_buffer注册的变量不能被optimizer.step更新,无论如何都不行,但是能被module.state_dict()识别到.

说明:此时只是记录一下,方便以后学习参考,好找~

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