Commit f7cf2bb0 authored by yinchimaoliang's avatar yinchimaoliang
Browse files

add gpu detection

parent df186989
import numpy as np
import pytest
import torch
from mmdet3d.datasets.scannet_dataset import ScannetDataset
......@@ -72,6 +73,8 @@ def test_getitem():
def test_evaluate():
if not torch.cuda.is_available():
pytest.skip()
root_path = './tests/data/scannet'
ann_file = './tests/data/scannet/scannet_infos.pkl'
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