Explorar el Código

update command.py

sunyanfang01 hace 5 años
padre
commit
c8c287fc76
Se han modificado 3 ficheros con 4 adiciones y 2 borrados
  1. 1 1
      paddlex/__init__.py
  2. 2 0
      paddlex/command.py
  3. 1 1
      setup.py

+ 1 - 1
paddlex/__init__.py

@@ -49,4 +49,4 @@ load_model = cv.models.load_model
 datasets = cv.datasets
 
 log_level = 2
-__version__ = '0.1.9.github'
+__version__ = '0.2.0.github'

+ 2 - 0
paddlex/command.py

@@ -71,6 +71,8 @@ def main():
             assert len(
                 fixed_input_shape
             ) == 2, "len of fixed input shape must == 2, such as [224,224]"
+        else:
+            fixed_input_shape = None
 
         model = pdx.load_model(args.model_dir, fixed_input_shape)
         model.export_inference_model(args.save_dir)

+ 1 - 1
setup.py

@@ -19,7 +19,7 @@ long_description = "PaddleX. A end-to-end deeplearning model development toolkit
 
 setuptools.setup(
     name="paddlex",
-    version='0.1.9',
+    version='0.2.0',
     author="paddlex",
     author_email="paddlex@baidu.com",
     description=long_description,