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",