new model

This commit is contained in:
adamnsandle
2020-12-22 12:14:54 +00:00
parent dee92fb847
commit b23720e3ca
4 changed files with 10 additions and 10 deletions

View File

@@ -233,7 +233,7 @@
" ort_inputs = {'input': inputs.cpu().numpy()}\n",
" outs = model.run(None, ort_inputs)\n",
" outs = [torch.Tensor(x) for x in outs]\n",
" return outs"
" return outs[0]"
]
},
{
@@ -405,5 +405,5 @@
}
},
"nbformat": 4,
"nbformat_minor": 0
"nbformat_minor": 1
}