The SurgicalProcedures.BN_to_GN is not work

[ghost]

Some BatchNormal layer in my neural network, which have been replaced by SurgicalProcedures.BN_to_GN. Now the deepee doesn't work. There is the error information:

deepee.snooper.BadModuleError: BatchNorm Layers must have track_running_stats turned off, otherwise be replaced w ith InstanceNorm, LayerNorm or GroupNorm.

can you tell me how to fix that problem?

[gkaissis]

Could you please give us some more information on how to reproduce your error? What exact model you are using etc.?

Thank you!