search

longcw / RoIAlign.pytorch

RoIAlign & crop_and_resize for PyTorch
555 stars 103 forks source link

error with test.py #5

Open xpain1 opened 6 years ago

xpain1 commented 6 years ago

Hi, When i run test.py, there is an error occured, as follows: F tensorflow/stream_executor/cuda/cuda_driver.cc:334] current context was not created by the StreamExecutor cuda_driver API: 0x3af31a0; a CUDA runtime call was likely performed without using a StreamExecutor context I only have a GPU ,what should i do ? Thanks

longcw commented 6 years ago

You can use cpu version of tensorflow if you only want to verify the result. I didn't test the script with tensorflow-gpu.

xpain1 commented 6 years ago

Hi, I run it correctly with your help,but when i read your code, i found that the transform_fpcoor False in line 128 is False. i think the transform_fpcoor should be True if you want to use roiAlign ,am i right? thanks