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 для любого модуля nn.BatchNorm в root

Тип возвращаемого значения: Модуль

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.func.replace_all_batch_norm_modules_.html