From 4c7b2bdc30657354afcbc255385a163662e66c8b Mon Sep 17 00:00:00 2001
From: rockstarr <41538890+miknyko@users.noreply.github.com>
Date: Thu, 25 Nov 2021 18:18:30 +0800
Subject: [PATCH] TorchScript `torch==1.7.0` Path support (#5781)

* fix path error in export.py

* Update export.py

updated!

* Update export.py

oops forget something
---
 export.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/export.py b/export.py
index 35875f1..9d6d049 100644
--- a/export.py
+++ b/export.py
@@ -71,7 +71,7 @@ def export_torchscript(model, im, file, optimize, prefix=colorstr('TorchScript:'
         ts = torch.jit.trace(model, im, strict=False)
         d = {"shape": im.shape, "stride": int(max(model.stride)), "names": model.names}
         extra_files = {'config.txt': json.dumps(d)}  # torch._C.ExtraFilesMap()
-        (optimize_for_mobile(ts) if optimize else ts).save(f, _extra_files=extra_files)
+        (optimize_for_mobile(ts) if optimize else ts).save(str(f), _extra_files=extra_files)
 
         LOGGER.info(f'{prefix} export success, saved as {f} ({file_size(f):.1f} MB)')
     except Exception as e: