lichen/ieemoo-ai-isempty
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update ieemoo-ai-isempty.py.

Browse Source
This commit is contained in:
Brainway
2022-10-11 07:38:14 +00:00
committed by Gitee
parent d5504fce91
commit 062e2245de
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ieemoo-ai-isempty.py
View File

@ -54,7 +54,7 @@ def parse_args(model_file="../module/ieemoo-ai-isempty/model/now/emptyjudge5_che
class Predictor(object): class Predictor(object):
def __init__(self, args): def __init__(self, args):
self.args = 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) #print(self.args.device)
self.args.nprocs = torch.cuda.device_count() self.args.nprocs = torch.cuda.device_count()
self.cls_dict = {} self.cls_dict = {}