longcw / RoIAlign.pytorch

RoIAlign & crop_and_resize for PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

error with test.py

xpain1 opened this issue · comments

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

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

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