-
Notifications
You must be signed in to change notification settings - Fork 20
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
Error when samples_per_gpu>1 #6
Comments
Thanks for your suggestion. The code does not support multiple samples per gpu now. The whole pipeline is quite long and queries in each sample differ a lot, so we need query padding in each module, which is quite complicated. Moreover, memory of most gpus cannot support larger samples. If your gpu memory is large enough, we strongly recommend using ResNet101 for stronger performance, which will comsume more gpu memory. |
Thank you for your kind reply! |
Thank you for your nice work!
I find that the training cannot run when setting samples_per_gpu>1 (line 231 of your configure file).
The error happens in line 440 of vips3d/models/attention_dert3d.py. I print the intermediate data and find this error comes from different shapes of img_meta['lidar2img'] in the same batch.
Can you kindly check your implementation? I believe it is important for running your code on large memory GPUs.
The text was updated successfully, but these errors were encountered: