pytorch中register_buffer的作用
原文链接:https://blog.youkuaiyun.com/u011622208/article/details/100042933例子:register_buffer("supports", torch.arange(Vmin, Vmax+DELTA_Z, DELTA_Z))向模块添加持久缓冲区。这通常用于注册不应被视为模型参数的缓冲区。例如,BatchNorm的running_mean不是...
转载
2019-12-13 14:30:30 ·
8872 阅读 ·
4 评论