Просмотр исходного кода

add option to disable half precision in test.py (#2507)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
5.0
Benjamin Fineran GitHub 3 лет назад
Родитель
Сommit
fd1679975b
Не найден GPG ключ соответствующий данной подписи Идентификатор GPG ключа: 4AEE18F83AFDEB23
1 измененных файлов: 2 добавлений и 1 удалений
  1. +2
    -1
      test.py

+ 2
- 1
test.py Просмотреть файл

@@ -37,6 +37,7 @@ def test(data,
plots=True,
wandb_logger=None,
compute_loss=None,
half_precision=True,
is_coco=False):
# Initialize/load model and set device
training = model is not None
@@ -61,7 +62,7 @@ def test(data,
# model = nn.DataParallel(model)

# Half
half = device.type != 'cpu' # half precision only supported on CUDA
half = device.type != 'cpu' and half_precision # half precision only supported on CUDA
if half:
model.half()


Загрузка…
Отмена
Сохранить