search

Hanqer / deep-hough-transform

Jittor and Pytorch code for paper "Deep Hough Transform for Semantic Line Detection" (ECCV 2020, PAMI 2021)
338 stars 71 forks source link

使用NKL数据集进行Test提示GPU内存不足 #44

Closed icelemon233 closed 2 years ago

icelemon233 commented 2 years ago

单块GPU处于空闲状态，且为一块RTX2080Ti。 提示： File "forward.py", line 81, in main total_time = test(test_loader, model, args) File "forward.py", line 102, in test key_points = model(images) …… …… RuntimeError: CUDA error: out of memory 请问如何解决？感谢。

HassanBinHaroon commented 2 years ago

@icelemon233 Take a look at the following repository too.

https://github.com/HassanBinHaroon/deep-hough-transform