Your code takes more than 5 hours under test conditions, can you provide a GPU test version

[summer918]

Your code takes more than 5 hours under test conditions, can you provide a GPU test version
self._cpu_device = torch.device("cuda" if torch.cuda.is_available() else "cpu") replace self._cpu_device = torch.device("cpu")
it isn't work?

[GuangxingHan]

It is already implemented in GPU. The CUDA environment is set up here https://github.com/facebookresearch/detectron2/blob/main/detectron2/engine/launch.py#L117