mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Add missing training flag to convert_sync_batchnorm

Browse Source
This commit is contained in:
Collin McCarthy 2025-01-21 11:41:48 -08:00 committed by Ross Wightman
parent cb4cea561a
commit 84631cb5c6
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
timm/layers/norm_act.py
View File

@ -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():