chore: renamed a notebook
Former-commit-id: 149fec5530bfa67ba2da3780e256d15a7fc1a8f1 [formerly 49e43d72aab9cdf237ac812fe51d6763465db4dd] Former-commit-id: 04330274d840746abe66b4eb4d09c9257de3ceed
This commit is contained in:
parent
1a94fa6c09
commit
ebb213c565
9
.vscode/settings.json
vendored
9
.vscode/settings.json
vendored
|
@ -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,
|
||||
}
|
||||
}
|
||||
|
|
|
@ -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",
|
Loading…
Reference in a new issue