feature extraction with docker: RuntimeWarning: invalid value encountered in greater_equal

[suzyahyah]

Hi, 
I was able to run the feature extraction script, scripts/ extract_imgfeat.sh with the Docker image from https://hub.docker.com/r/chenrocks/butd-caffe/tags?page=1&ordering=last_updated

It processed ~750 images before I encountered this:

/src/tools/../lib/rpn/proposal_layer.py:184: RuntimeWarning: invalid value encountered in greater_equal keep = np.where((ws >= min_size) & (hs >= min_size))[0] F0607 05:45:37.953456 123 roi_pooling_layer.cu:87] Check failed: error == cudaSuccess (9 vs. 0) invalid configuration argument

Do you have any pointers on what I could possibly try to fix this? Thank you!