hatcher/MPKFNet
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

first commit

Browse Source
This commit is contained in:
hatcher committed 2026-07-17 18:55:31 +08:00
1 parent ae53e9eac5
commit 91cbd01126
514 files changed
+55625 -2203
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

DclTest.py
+1 -1
View File
@@ -44,7 +44,7 @@ from Dcl.models.LoadModel import MainModel
from Dcl.DclConfig import LoadConfig, load_data_transformers
from Dcl.util.test_tool import set_text, save_multi_img, cls_base_acc
os.environ['CUDA_VISIBLE_DEVICES'] = '6'
#os.environ['CUDA_VISIBLE_DEVICES'] = '6'
devices = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
class DclInference: