Searched refs:_replace_batchnorm (Results 1 – 1 of 1) sorted by relevance
92 def _replace_batchnorm(m: torch.fx.GraphModule, train_to_eval: bool): function182 _replace_batchnorm(model, train_to_eval=True)194 _replace_batchnorm(model, train_to_eval=False)