From 84631cb5c6c2b56d2ef65101fc916aa87fa83cd6 Mon Sep 17 00:00:00 2001 From: Collin McCarthy Date: Tue, 21 Jan 2025 11:41:48 -0800 Subject: [PATCH] Add missing training flag to convert_sync_batchnorm --- timm/layers/norm_act.py | 1 + 1 file changed, 1 insertion(+) diff --git a/timm/layers/norm_act.py b/timm/layers/norm_act.py index 496efcfd..f2117437 100644 --- a/timm/layers/norm_act.py +++ b/timm/layers/norm_act.py @@ -176,6 +176,7 @@ def convert_sync_batchnorm(module, process_group=None): module_output.running_mean = module.running_mean module_output.running_var = module.running_var module_output.num_batches_tracked = module.num_batches_tracked + module_output.training = module.training if hasattr(module, "qconfig"): module_output.qconfig = module.qconfig for name, child in module.named_children():