pyesonekyaw
commited on
Commit
·
550063f
1
Parent(s):
aa3fc43
cpu
Browse files- hubconf.py +1 -1
hubconf.py
CHANGED
@@ -40,7 +40,7 @@ def _create(name, pretrained=True, channels=3, classes=80, autoshape=True, verbo
|
|
40 |
name = Path(name)
|
41 |
path = name.with_suffix('.pt') if name.suffix == '' else name # checkpoint path
|
42 |
try:
|
43 |
-
device =
|
44 |
#device = 'mps'
|
45 |
if pretrained and channels == 3 and classes == 80:
|
46 |
model = DetectMultiBackend(path, device=device) # download/load FP32 model
|
|
|
40 |
name = Path(name)
|
41 |
path = name.with_suffix('.pt') if name.suffix == '' else name # checkpoint path
|
42 |
try:
|
43 |
+
device = 'cpu'
|
44 |
#device = 'mps'
|
45 |
if pretrained and channels == 3 and classes == 80:
|
46 |
model = DetectMultiBackend(path, device=device) # download/load FP32 model
|