saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-23 19:32:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Last orttraining build fix...

Browse source
This commit is contained in:
Ryan Hill 2021-05-10 16:52:51 -07:00
parent 149c037a4a
commit dfc8dab1b9
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
orttraining/orttraining/models/mnist/main.cc
View file

@ -176,6 +176,7 @@ Status ParseArguments(int argc, char* argv[], MnistParameters& params) {
0,
true,
0,
nullptr,
nullptr};
params.providers.emplace(kCudaExecutionProvider, CreateExecutionProviderFactory_Cuda(&info));