diff --git a/model_zoo/densenet121/src/utils/var_init.py b/model_zoo/densenet121/src/utils/var_init.py index 0512c7d6ae4..3791dc08be6 100644 --- a/model_zoo/densenet121/src/utils/var_init.py +++ b/model_zoo/densenet121/src/utils/var_init.py @@ -127,6 +127,9 @@ class KaimingInit(init.Initializer): self.mode = mode self.gain = _calculate_gain(nonlinearity, a) + def _initialize(self, arr): + pass + class KaimingUniform(KaimingInit): r"""