不只是nn.Module?ema-pytorch隐藏的Pytree特性让任意张量树也能做EMA跟踪
不只是nn.Module?ema-pytorch隐藏的Pytree特性让任意张量树也能做EMA跟踪 【免费下载链接】ema-pytorch A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model 项目地址: https://gitcode.com/gh_mirrors/em/ema-py…
2026/8/24 8:19:53