fix minor typo.

Signed-off-by: Ranganath Krishnan <ranganath.krishnan@intel.com>
IntelLabs · Oct 5, 2021 · 872d379 · 872d379
1 parent 782820a
commit 872d379
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bayesian_torch/utils/util.py b/bayesian_torch/utils/util.py
@@ -86,7 +86,7 @@ def MOPED(model, det_model, det_checkpoint, delta):
     for (idx, layer), (det_idx,
                        det_layer) in zip(enumerate(model.modules()),
                                          enumerate(det_model.modules())):
-        if (str(layer) == 'Conv1dReparametrization()'
+        if (str(layer) == 'Conv1dReparameterization()'
                 or str(layer) == 'Conv2dReparameterization()'
                 or str(layer) == 'Conv3dReparameterization()'
                 or str(layer) == 'ConvTranspose1dReparameterization()'