手机随时阅读
新人专享大礼包¥24
pytorch中的nn.Module.register_buffer及DDP模块的一些小知识 作用:给一个nn的Module添加一个buffer,而且不会被认为是一个model的parameter。BN的running_mean就是一个buffer。这个buffer会被para一起,存储到模...
Copyright © 2005-2022 51CTO.COM 版权所有 京ICP证060544号