We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 75fe5a3 commit 8dd3808Copy full SHA for 8dd3808
torch/_functorch/batch_norm_replacement.py
@@ -1,10 +1,8 @@
1
-# mypy: allow-untyped-decorators
2
-# mypy: allow-untyped-defs
3
import torch.nn as nn
4
from torch._functorch.utils import exposed_in
5
6
7
-def batch_norm_without_running_stats(module: nn.Module):
+def batch_norm_without_running_stats(module: nn.Module) -> None:
8
if (
9
isinstance(module, nn.modules.batchnorm._BatchNorm)
10
and module.track_running_stats
0 commit comments