Searched refs:replace_all_batch_norm_modules_ (Results 1 – 9 of 9) sorted by relevance
18 from functorch.experimental import replace_all_batch_norm_modules_20 replace_all_batch_norm_modules_(model)45 from functorch.experimental import replace_all_batch_norm_modules_47 replace_all_batch_norm_modules_(model)86 from functorch.experimental import replace_all_batch_norm_modules_88 replace_all_batch_norm_modules_(model)
66 from functorch.experimental import replace_all_batch_norm_modules_68 replace_all_batch_norm_modules_(model)
42 have a net that you want to change, you can run ``replace_all_batch_norm_modules_`` to update the47 from functorch.experimental import replace_all_batch_norm_modules_48 replace_all_batch_norm_modules_(net)
68 ``replace_all_batch_norm_modules_`` to update the module in-place to not use running stats72 from torch.func import replace_all_batch_norm_modules_73 replace_all_batch_norm_modules_(net)
70 replace_all_batch_norm_modules_
4 from torch._functorch.batch_norm_replacement import replace_all_batch_norm_modules_
12 from torch._functorch.batch_norm_replacement import replace_all_batch_norm_modules_
19 def replace_all_batch_norm_modules_(root: nn.Module) -> nn.Module: function
41 from functorch.experimental import functionalize, replace_all_batch_norm_modules_4258 replace_all_batch_norm_modules_(net)