From 208bdc0570ec6d7cda3e96edfe78fbcebfcab9ed Mon Sep 17 00:00:00 2001
From: Mark Kurtz <mark@neuralmagic.com>
Date: Wed, 12 May 2021 18:28:25 -0400
Subject: [PATCH] Pytorch manager modify fixes

---
 src/sparseml/pytorch/optim/manager.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/sparseml/pytorch/optim/manager.py b/src/sparseml/pytorch/optim/manager.py
index 3168aee3e73..89e7561c610 100644
--- a/src/sparseml/pytorch/optim/manager.py
+++ b/src/sparseml/pytorch/optim/manager.py
@@ -368,13 +368,13 @@ def modify(
             epoch = self._initialize_epoch
 
         if not self.initialized:
-            self.intialize(module, epoch)
+            self.initialize(module, epoch)
 
         if wrap_optim is None:
             wrap_optim = optimizer
 
         return RecipeManagerStepWrapper(
-            wrap_optim, self, optimizer, module, epoch, steps_per_epoch
+            wrap_optim, optimizer, module, self, epoch, steps_per_epoch
         )
 
     def finalize(