pyesonekyaw commited on
Commit
550063f
1 Parent(s): aa3fc43
Files changed (1) hide show
  1. 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 = select_device(('0' if torch.cuda.is_available() else 'cpu') if device is None else 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