Преглед на файлове

set default output of clas pipeline

gaotingquan преди 1 година
родител
ревизия
5a9bf15240
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      paddlex/pipelines/image_classification/pipeline.py

+ 1 - 1
paddlex/pipelines/image_classification/pipeline.py

@@ -26,7 +26,7 @@ class ClsPipeline(BasePipeline):
     def __init__(self,
                  model_name=None,
                  model_dir=None,
-                 output=None,
+                 output="./output",
                  kernel_option=None,
                  device="gpu",
                  **kwargs):