tensorplay.func.replace_all_batch_norm_modules_

tensorplay.func.replace_all_batch_norm_modules_(root: Module) → Module

Drops the running statistics of every batch-normalization module in root, in place, and returns root.