Allow loading from model files with empty weights. (#16061)

* Allow loading from model files with empty weights. * ci * ci * CI
apache · Nov 26, 2019 · 6b00b2c · 6b00b2c
1 parent a11b7ea
commit 6b00b2c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/mxnet/model.py b/python/mxnet/model.py
@@ -431,6 +431,7 @@ def load_params(prefix, epoch):
     aux_params = {}
     if not save_dict:
         logging.warning("Params file '%s' is empty", '%s-%04d.params' % (prefix, epoch))
+        return (arg_params, aux_params)
     for k, v in save_dict.items():
         tp, name = k.split(":", 1)
         if tp == "arg":