torch.func.replace_all_batch_norm_modules_

torch.func.replace_all_batch_norm_modules_(root)

透過將 running_mean 和 running_var 設定為 None，並將 track_running_stats 設定為 False，以就地更新 root，適用於 root 中的任何 nn.BatchNorm 模組

回傳類型

Module