ENSEEIHT/projet-long
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

chore: renamed a notebook

Browse source 
Former-commit-id: 149fec5530bfa67ba2da3780e256d15a7fc1a8f1 [formerly 49e43d72aab9cdf237ac812fe51d6763465db4dd]
Former-commit-id: 04330274d840746abe66b4eb4d09c9257de3ceed
This commit is contained in:
Laurent Fainsin 2022-07-08 13:35:24 +02:00
parent 1a94fa6c09
commit ebb213c565
2 changed files with 11 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
.vscode/settings.json vendored
View file

@ -10,4 +10,13 @@
"source.organizeImports": true "source.organizeImports": true
} }
}, },
"files.exclude": {
"**/.git": true,
"**/.svn": true,
"**/.hg": true,
"**/CVS": true,
"**/.DS_Store": true,
"**/Thumbs.db": true,
"**/__pycache__": true,
}
} }

4
src/dynamic.ipynb → src/convert.ipynb
View file

@ -43,7 +43,7 @@
], ],
"source": [ "source": [
"net.load_state_dict(\n", "net.load_state_dict(\n",
" torch.load(\"../best.pth\")\n", " torch.load(\"../checkpoint/best.pth\")\n",
")\n" ")\n"
] ]
}, },
@ -57,7 +57,7 @@
"torch.onnx.export(\n", "torch.onnx.export(\n",
" net,\n", " net,\n",
" dummy_input,\n", " dummy_input,\n",
" \"model-test.onnx\",\n", " \"../checkpoint/best.onnx\",\n",
" opset_version=14,\n", " opset_version=14,\n",
" input_names=[\"input\"],\n", " input_names=[\"input\"],\n",
" output_names=[\"output\"],\n", " output_names=[\"output\"],\n",