Skip to content

GitLab

Commit f7cf2bb0 authored by yinchimaoliang's avatar yinchimaoliang
Browse files

add gpu detection

parent df186989
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
tests/test_scannet_dataset.py
View file @ f7cf2bb0
import numpy as np import numpy as np
import pytest
import torch import torch
from mmdet3d.datasets.scannet_dataset import ScannetDataset from mmdet3d.datasets.scannet_dataset import ScannetDataset
...@@ -72,6 +73,8 @@ def test_getitem(): ...@@ -72,6 +73,8 @@ def test_getitem():
def test_evaluate(): def test_evaluate():
if not torch.cuda.is_available():
pytest.skip()
root_path = './tests/data/scannet' root_path = './tests/data/scannet'
ann_file = './tests/data/scannet/scannet_infos.pkl' ann_file = './tests/data/scannet/scannet_infos.pkl'
scannet_dataset = ScannetDataset(root_path, ann_file) scannet_dataset = ScannetDataset(root_path, ann_file)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment