supervisely 6.73.339__py3-none-any.whl → 6.73.340__py3-none-any.whl

supervisely/nn/inference/inference.py

@@ -598,15 +598,16 @@ class Inference:
     def _download_model_files(self, deploy_params: dict, log_progress: bool = True) -> dict:
         if deploy_params["runtime"] != RuntimeType.PYTORCH:
             export = deploy_params["model_info"].get("export", {})
-            export_model = export.get(deploy_params["runtime"], None)
-            if export_model is not None:
-                if sly_fs.get_file_name(export_model) == sly_fs.get_file_name(
-                    deploy_params["model_files"]["checkpoint"]
-                ):
-                    deploy_params["model_files"]["checkpoint"] = (
-                        deploy_params["model_info"]["artifacts_dir"] + export_model
-                    )
-                    logger.info(f"Found model checkpoint for '{deploy_params['runtime']}'")
+            if export is not None:
+                export_model = export.get(deploy_params["runtime"], None)
+                if export_model is not None:
+                    if sly_fs.get_file_name(export_model) == sly_fs.get_file_name(
+                        deploy_params["model_files"]["checkpoint"]
+                    ):
+                        deploy_params["model_files"]["checkpoint"] = (
+                            deploy_params["model_info"]["artifacts_dir"] + export_model
+                        )
+                        logger.info(f"Found model checkpoint for '{deploy_params['runtime']}'")
 
         if deploy_params["model_source"] == ModelSource.PRETRAINED:
             return self._download_pretrained_model(deploy_params["model_files"], log_progress)

supervisely/nn/training/train_app.py

@@ -2524,17 +2524,26 @@ class TrainApp:
             self.gui.hyperparameters_selector.get_export_onnx_checkbox_value() is True
             and self._convert_onnx_func is not None
         ):
-            self._set_text_status("export_onnx")
-            onnx_path = self._convert_onnx_func(experiment_info)
-            export_weights[RuntimeType.ONNXRUNTIME] = onnx_path
+            try:
+                self._set_text_status("export_onnx")
+                onnx_path = self._convert_onnx_func(experiment_info)
+                if sly_fs.file_exists(onnx_path):
+                    export_weights[RuntimeType.ONNXRUNTIME] = onnx_path
+            except Exception as e:
+                logger.error(f"Failed to export ONNX model: {e}")
 
         if (
             self.gui.hyperparameters_selector.get_export_tensorrt_checkbox_value() is True
             and self._convert_tensorrt_func is not None
         ):
-            self._set_text_status("export_trt")
-            tensorrt_path = self._convert_tensorrt_func(experiment_info)
-            export_weights[RuntimeType.TENSORRT] = tensorrt_path
+            try:
+                self._set_text_status("export_trt")
+                tensorrt_path = self._convert_tensorrt_func(experiment_info)
+                if sly_fs.file_exists(tensorrt_path):
+                    export_weights[RuntimeType.TENSORRT] = tensorrt_path
+            except Exception as e:
+                logger.error(f"Failed to export TensorRT model: {e}")
+                
         return export_weights
 
     def _upload_export_weights(

{supervisely-6.73.339.dist-info → supervisely-6.73.340.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: supervisely
-Version: 6.73.339
+Version: 6.73.340
 Summary: Supervisely Python SDK.
 Home-page: https://github.com/supervisely/supervisely
 Author: Supervisely

{supervisely-6.73.339.dist-info → supervisely-6.73.340.dist-info}/RECORD

@@ -883,7 +883,7 @@ supervisely/nn/benchmark/visualization/widgets/table/__init__.py,sha256=47DEQpj8
 supervisely/nn/benchmark/visualization/widgets/table/table.py,sha256=atmDnF1Af6qLQBUjLhK18RMDKAYlxnsuVHMSEa5a-e8,4319
 supervisely/nn/inference/__init__.py,sha256=QFukX2ip-U7263aEPCF_UCFwj6EujbMnsgrXp5Bbt8I,1623
 supervisely/nn/inference/cache.py,sha256=LAirR5mFHCtK59EO1lefQ2qhpp0vBvRTH26EVrs13Y0,32073
-supervisely/nn/inference/inference.py,sha256=dzfvHjisVYgsijnHYazb_y0mXYZQJX0E1fEsDY6S5lw,169730
+supervisely/nn/inference/inference.py,sha256=FUTlgyy676IsFhNH6PTRdnGz6aOXXH_5RtRKqxX8Pss,169801
 supervisely/nn/inference/session.py,sha256=jmkkxbe2kH-lEgUU6Afh62jP68dxfhF5v6OGDfLU62E,35757
 supervisely/nn/inference/video_inference.py,sha256=8Bshjr6rDyLay5Za8IB8Dr6FURMO2R_v7aELasO8pR4,5746
 supervisely/nn/inference/gui/__init__.py,sha256=wCxd-lF5Zhcwsis-wScDA8n1Gk_1O00PKgDviUZ3F1U,221
@@ -980,7 +980,7 @@ supervisely/nn/tracker/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NM
 supervisely/nn/tracker/utils/gmc.py,sha256=3JX8979H3NA-YHNaRQyj9Z-xb9qtyMittPEjGw8y2Jo,11557
 supervisely/nn/tracker/utils/kalman_filter.py,sha256=eSFmCjM0mikHCAFvj-KCVzw-0Jxpoc3Cfc2NWEjJC1Q,17268
 supervisely/nn/training/__init__.py,sha256=gY4PCykJ-42MWKsqb9kl-skemKa8yB6t_fb5kzqR66U,111
-supervisely/nn/training/train_app.py,sha256=mUEXGE9rQZKyVCrX1qjyDFOEuKqdVVa5jQ0EnP0gBGI,106273
+supervisely/nn/training/train_app.py,sha256=GsE8uh_PnExFceEyrmy0ubccbgSf96KCToyP8BXodI0,106666
 supervisely/nn/training/gui/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 supervisely/nn/training/gui/classes_selector.py,sha256=Bpp-RFDQqcZ0kLJmS6ZnExkdscWwRusvF4vbWjEsKlQ,3926
 supervisely/nn/training/gui/gui.py,sha256=QDeWa6iow3z3EwLZcHlQ48d4Ur6hCmedNvCod-8qZ-k,27978
@@ -1082,9 +1082,9 @@ supervisely/worker_proto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZ
 supervisely/worker_proto/worker_api_pb2.py,sha256=VQfi5JRBHs2pFCK1snec3JECgGnua3Xjqw_-b3aFxuM,59142
 supervisely/worker_proto/worker_api_pb2_grpc.py,sha256=3BwQXOaP9qpdi0Dt9EKG--Lm8KGN0C5AgmUfRv77_Jk,28940
 supervisely_lib/__init__.py,sha256=7-3QnN8Zf0wj8NCr2oJmqoQWMKKPKTECvjH9pd2S5vY,159
-supervisely-6.73.339.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-supervisely-6.73.339.dist-info/METADATA,sha256=jUY-G7GqpZ9Qe4qSdlGrj8lMGoV8P386z114kCWh7gk,33596
-supervisely-6.73.339.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-supervisely-6.73.339.dist-info/entry_points.txt,sha256=U96-5Hxrp2ApRjnCoUiUhWMqijqh8zLR03sEhWtAcms,102
-supervisely-6.73.339.dist-info/top_level.txt,sha256=kcFVwb7SXtfqZifrZaSE3owHExX4gcNYe7Q2uoby084,28
-supervisely-6.73.339.dist-info/RECORD,,
+supervisely-6.73.340.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+supervisely-6.73.340.dist-info/METADATA,sha256=MYPQWfGjiMVfYMHxrVyPYtnuYO1c9Hqa1l4PcmRU51k,33596
+supervisely-6.73.340.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+supervisely-6.73.340.dist-info/entry_points.txt,sha256=U96-5Hxrp2ApRjnCoUiUhWMqijqh8zLR03sEhWtAcms,102
+supervisely-6.73.340.dist-info/top_level.txt,sha256=kcFVwb7SXtfqZifrZaSE3owHExX4gcNYe7Q2uoby084,28
+supervisely-6.73.340.dist-info/RECORD,,