Skip to content

Commit f05418f

Browse files
author
yixu.cui
committed
tiny changes
1 parent 8c5ddb5 commit f05418f

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

utils/google_utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,8 @@ def gsutil_getsize(url=''):
1818

1919
def attempt_download(file, repo='WongKinYiu/yolov7'):
2020
# Attempt file download if does not exist
21-
file = Path(str(file).strip().replace("'", '').lower())
21+
# file = Path(str(file).strip().replace("'", '').lower())
22+
file = Path(str(file).strip().replace("'", ''))
2223

2324
if not file.exists():
2425
try:

utils/torch_utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,8 @@ def select_device(device='', batch_size=None):
8383
s += 'CPU\n'
8484

8585
logger.info(s.encode().decode('ascii', 'ignore') if platform.system() == 'Windows' else s) # emoji-safe
86-
return torch.device('cuda:0' if cuda else 'cpu')
86+
# return torch.device('cuda:0' if cuda else 'cpu')
87+
return torch.device('cuda' if cuda else 'cpu') # [cui] what's the difference between 'cuda:0' and 'cuda'
8788

8889

8990
def time_synchronized():

0 commit comments

Comments
 (0)