register_buffer

郭庆汝1 年前
人工智能·深度学习·register_buffer
深度学习基础知识 register_buffer 与 register_parameter用法分析思考问题：定义的weight与bias是否会被保存到网络的参数中，可否在优化器的作用下进行学习验证方案：定义网络模型，设置weigut与bias，遍历网络结构参数net.named_parameters()，如果定义的weight与bias在里面，则说明是可学习参数；否则，是不可学习参数