update ieemoo-ai-isempty.py.

2022-10-11 07:38:14 +00:00
parent d5504fce91
commit 062e2245de
1 changed files with 2 additions and 2 deletions
--- a/ieemoo-ai-isempty.py
+++ b/ieemoo-ai-isempty.py
@ -54,7 +54,7 @@ def parse_args(model_file="../module/ieemoo-ai-isempty/model/now/emptyjudge5_che
 class Predictor(object):
    def __init__(self, args):
        self.args = args
-        self.args.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+        self.args.device = torch.device("cpu")
        #print(self.args.device)
        self.args.nprocs = torch.cuda.device_count()
        self.cls_dict = {}