From ee6bc1cea24556f603b75f2776f9b0270c184995 Mon Sep 17 00:00:00 2001 From: rockmastar <42338705@qq.com> Date: Thu, 25 Nov 2021 15:16:42 +0800 Subject: [PATCH 1/2] fix the path error in export.py --- export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/export.py b/export.py index 2aca0f341dbd..dffa2fb4bebf 100644 --- a/export.py +++ b/export.py @@ -54,7 +54,7 @@ def export_torchscript(model, im, file, optimize, prefix=colorstr('TorchScript:' f = file.with_suffix('.torchscript.pt') ts = torch.jit.trace(model, im, strict=False) - (optimize_for_mobile(ts) if optimize else ts).save(f) + (optimize_for_mobile(ts) if optimize else ts).save(f.as_posix()) print(f'{prefix} export success, saved as {f} ({file_size(f):.1f} MB)') except Exception as e: From 7ca7579aaee63f40a767cdebcb725955e4a0c85e Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 25 Nov 2021 08:40:55 +0100 Subject: [PATCH 2/2] Update export.py --- export.py | 1 - 1 file changed, 1 deletion(-) diff --git a/export.py b/export.py index a891e2f9322e..35875f1fb0d3 100644 --- a/export.py +++ b/export.py @@ -73,7 +73,6 @@ def export_torchscript(model, im, file, optimize, prefix=colorstr('TorchScript:' extra_files = {'config.txt': json.dumps(d)} # torch._C.ExtraFilesMap() (optimize_for_mobile(ts) if optimize else ts).save(f, _extra_files=extra_files) - LOGGER.info(f'{prefix} export success, saved as {f} ({file_size(f):.1f} MB)') except Exception as e: LOGGER.info(f'{prefix} export failure: {e}')