소스 검색

update ts export.py

Sunflower7788 1 년 전
부모
커밋
48254c8a3e
3개의 변경된 파일1개의 추가작업 그리고 4개의 파일을 삭제
  1. 0 1
      paddlex/modules/ts_anomaly_detection/exportor.py
  2. 1 2
      paddlex/modules/ts_classification/exportor.py
  3. 0 1
      paddlex/modules/ts_forecast/exportor.py

+ 0 - 1
paddlex/modules/ts_anomaly_detection/exportor.py

@@ -48,4 +48,3 @@ class TSADExportor(BaseExportor):
             self.export_config.weight_path = dest_path.joinpath(
                 "best_accuracy.pdparams/best_model/model.pdparams"
             )
-            

+ 1 - 2
paddlex/modules/ts_classification/exportor.py

@@ -38,7 +38,7 @@ class TSCLSExportor(BaseExportor):
         self.uncompress_tar_file()
         config_path = Path(self.export_config.weight_path).parent.parent / "config.yaml"
         return config_path
-    
+
     def uncompress_tar_file(self):
         """unpackage the tar file containing training outputs and update weight path"""
         if tarfile.is_tarfile(self.export_config.weight_path):
@@ -48,4 +48,3 @@ class TSCLSExportor(BaseExportor):
             self.export_config.weight_path = dest_path.joinpath(
                 "best_accuracy.pdparams/best_model/model.pdparams"
             )
-    

+ 0 - 1
paddlex/modules/ts_forecast/exportor.py

@@ -48,4 +48,3 @@ class TSFCExportor(BaseExportor):
             self.export_config.weight_path = dest_path.joinpath(
                 "best_accuracy.pdparams/best_model/model.pdparams"
             )
-