Merge pull request #155 from ShivamKumar2002/patch-1
Fix GPU not being used even when it available
This commit is contained in:
commit
6416d4ebe8
@ -1,3 +1,4 @@
|
|||||||
|
import torch
|
||||||
import onnxruntime
|
import onnxruntime
|
||||||
|
|
||||||
use_gpu = False
|
use_gpu = False
|
||||||
|
Loading…
Reference in New Issue
Block a user