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

GuangxingHan / FCT

Code for CVPR 2022 Oral paper: 'Few-Shot Object Detection with Fully Cross-Transformer'

72 stars 13 forks source link

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

Open summer918 opened 10 months ago

summer918 commented 10 months ago

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 commented 10 months ago

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