Skip to content

GitLab

Commit e6cb749b authored by Youngjin Kim's avatar Youngjin Kim Committed by mcarilli
Browse files

Fixed error in convert_syncbn_model function (#380)

parent ad98cc5f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
apex/parallel/__init__.py
View file @ e6cb749b
...@@ -43,6 +43,7 @@ def convert_syncbn_model(module, process_group=None, channel_last=False): ...@@ -43,6 +43,7 @@ def convert_syncbn_model(module, process_group=None, channel_last=False):
mod = SyncBatchNorm(module.num_features, module.eps, module.momentum, module.affine, module.track_running_stats, process_group, channel_last=channel_last) mod = SyncBatchNorm(module.num_features, module.eps, module.momentum, module.affine, module.track_running_stats, process_group, channel_last=channel_last)
mod.running_mean = module.running_mean mod.running_mean = module.running_mean
mod.running_var = module.running_var mod.running_var = module.running_var
mod.num_batches_tracked = module.num_batches_tracked
if module.affine: if module.affine:
mod.weight.data = module.weight.data.clone().detach() mod.weight.data = module.weight.data.clone().detach()
mod.bias.data = module.bias.data.clone().detach() mod.bias.data = module.bias.data.clone().detach()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment