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