-
Notifications
You must be signed in to change notification settings - Fork 47
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
for help #2
Comments
HI @pengzhou1108 , I have the similar issues and my GPU is GeForce 1080, ~8G. After changing RPN batch size to 1, still doesn't work. Any ideas to figure this out? Issues in log: E tensorflow/stream_executor/cuda/cuda_fft.cc:169] failed to create cuFFT batched plan:2 During handling of the above exception, another exception occurred: Traceback (most recent call last): Caused by op 'noise_pred/FFT', defined at: InternalError (see above for traceback): c2c fft failed : in.shape=[3136,16384] |
Hi, I guess you did not change the batch size correctly. How do you change the RPN batch size? You should change the number in cfgs/*.yml file instead of the files in lib folder, which is the actual parameter used for training and testing. |
I set TRAIN.RPN_BATCHSIZE = 1 in train_faster_rcnn.sh to override the config. In the printed log I noticed the setting worked. One update: rois, cls_prob, bbox_pred = self.build_network(sess, training) |
In the log figure you sent indicates the the dimensionality of bilinear layer is [3136(64X7x7), 16384], which has 64 as batch size (the default RPN batch size in yml file). In my case sequential is set to be False. |
En...Thanks! I will change the config yml directly. |
Sorry to disturb you again,How to solve this problem?

The text was updated successfully, but these errors were encountered: