2019年10月31日
register_buffer vs register_parameter
摘要： 先来看下 的成员： 和`register_parameter _buffer _parameters`，调用这两个函数分别会向两个成员写入数据。 和`_parameter state_dict .cpu() .cuda()`在设备间进行转换。 中的元素不会被优化器更新，如果在模型中需要需要一些参数， 阅读全文
posted @ 2019-10-31 21:10 骏腾 阅读(4392) 评论(5) 推荐(0)
pytorch nn.Parameters vs nn.Module.register_parameter
摘要： nn.Parameters vs nn.Module.register_parameter 阅读全文
posted @ 2019-10-31 20:38 骏腾 阅读(5003) 评论(0) 推荐(2)