dict-guided icon indicating copy to clipboard operation
dict-guided copied to clipboard

Important fix for cpu usage in adet/modeling/attn_predictor.py

Open doanthuyennt opened this issue 3 years ago • 0 comments

Hello again Mr author, There is a bug I came a cross when I use only cpu for predicting, image What my approach was to change every to.("cuda") in to to(rois.device) ## Since I want to be consistent with your code anyway. image After that, run python setup.py install -v again ... and voila it worked! image

Tks for reading this.

doanthuyennt avatar Sep 21 '21 15:09 doanthuyennt