torch.func.replace_all_batch_norm_modules_¶

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

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