torch.func.replace_all_batch_norm_modules_¶

torch.func.replace_all_batch_norm_modules_(root)[源代码]¶

通过将 root 中任何 nn.BatchNorm 模块的 running_mean 和 running_var 设置为 None 并将 track_running_stats 设置为 False，就地更新 root