Table of Contents

Shortcuts

torch.func.replace_all_batch_norm_modules_¶

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

In place updates root by setting the running_mean and running_var to be None and setting track_running_stats to be False for any nn.BatchNorm module in root

Return type: Module

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources