Fix unused arg

apache · Aug 8, 2019 · 306e2df · 306e2df
1 parent 919879d
commit 306e2df
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/mxnet/module/bucketing_module.py b/python/mxnet/module/bucketing_module.py
@@ -581,7 +581,7 @@ def save_checkpoint(self, prefix, epoch, remove_amp_cast=False):
         nd.save("%s.buckets" % (prefix), nd.array(list(self._buckets.keys()), dtype=np.int32))
 
     @staticmethod
-    def load(prefix, epoch, load_optimizer_states=False, sym_gen=None, default_bucket_key=None, **kwargs):
+    def load(prefix, epoch, sym_gen=None, default_bucket_key=None, **kwargs):
         """Creates a model from previously saved checkpoint.
 
         Parameters