xiaohaoChen / rrc_detection

Accurate Single Stage Detector Using Recurrent Rolling Convolution
Other
361 stars 136 forks source link

GPU #33

Closed YangBain closed 7 years ago

YangBain commented 7 years ago

@jimmy-ren Hello,

This work doesn't support NVIDIA 1080, isn't it?

Thank you very much.

Bin Yang

xiaohaoChen commented 7 years ago

This code can run on a Nvidia GTX 1080. But you should reduce the rolling_time and the branch_num, as GTX 1080 has only 8GB memory.

YangBain commented 7 years ago

Thank you very much. It works now.