reverse providers logic
This commit is contained in:
@@ -1,2 +1,4 @@
|
|||||||
|
import onnxruntime
|
||||||
|
|
||||||
use_gpu = False
|
use_gpu = False
|
||||||
providers = ['CPUExecutionProvider']
|
providers = onnxruntime.get_available_providers()
|
||||||
|
|||||||
5
run.py
5
run.py
@@ -1,9 +1,8 @@
|
|||||||
import sys
|
import sys
|
||||||
import onnxruntime
|
|
||||||
import core.globals
|
import core.globals
|
||||||
|
|
||||||
if '--gpu' in sys.argv:
|
if '--gpu' not in sys.argv:
|
||||||
core.globals.providers = onnxruntime.get_available_providers()
|
core.globals.providers = ['CPUExecutionProvider']
|
||||||
|
|
||||||
import glob
|
import glob
|
||||||
import argparse
|
import argparse
|
||||||
|
|||||||
Reference in New Issue
Block a user