From ebb213c565f86762042c6467272c1945a61d5c1d Mon Sep 17 00:00:00 2001 From: Laurent Fainsin Date: Fri, 8 Jul 2022 13:35:24 +0200 Subject: [PATCH] chore: renamed a notebook Former-commit-id: 149fec5530bfa67ba2da3780e256d15a7fc1a8f1 [formerly 49e43d72aab9cdf237ac812fe51d6763465db4dd] Former-commit-id: 04330274d840746abe66b4eb4d09c9257de3ceed --- .vscode/settings.json | 9 +++++++++ src/{dynamic.ipynb => convert.ipynb} | 4 ++-- 2 files changed, 11 insertions(+), 2 deletions(-) rename src/{dynamic.ipynb => convert.ipynb} (95%) diff --git a/.vscode/settings.json b/.vscode/settings.json index 6e422f4..252b13f 100644 --- a/.vscode/settings.json +++ b/.vscode/settings.json @@ -10,4 +10,13 @@ "source.organizeImports": true } }, + "files.exclude": { + "**/.git": true, + "**/.svn": true, + "**/.hg": true, + "**/CVS": true, + "**/.DS_Store": true, + "**/Thumbs.db": true, + "**/__pycache__": true, + } } diff --git a/src/dynamic.ipynb b/src/convert.ipynb similarity index 95% rename from src/dynamic.ipynb rename to src/convert.ipynb index 371cb92..94363d6 100644 --- a/src/dynamic.ipynb +++ b/src/convert.ipynb @@ -43,7 +43,7 @@ ], "source": [ "net.load_state_dict(\n", - " torch.load(\"../best.pth\")\n", + " torch.load(\"../checkpoint/best.pth\")\n", ")\n" ] }, @@ -57,7 +57,7 @@ "torch.onnx.export(\n", " net,\n", " dummy_input,\n", - " \"model-test.onnx\",\n", + " \"../checkpoint/best.onnx\",\n", " opset_version=14,\n", " input_names=[\"input\"],\n", " output_names=[\"output\"],\n",