-
Notifications
You must be signed in to change notification settings - Fork 136
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
hi,now i am facing a problem when i ran ./run_test.sh #18
Comments
It is exactly as the msg said: CUDA out of memory |
Thanks! i have solved this problem by change a GPU with 9GB! |
hi ,I run mono_3d_tracking.py Process finished with exit code 1I find the code can iter 2 times normally ,then it broken ,can you give me some advice to solve it??? for i, (image, box_info) in enumerate(iter(test_loader)):
======================================================= |
File "test_net.py", line 304, in
RCNN_loss_center, rois_label = fasterRCNN(im_data, im_info, gt_boxes,num_boxes, fixed_center)
File "/home/ld/.conda/envs/dave/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/ld/3d-vehicle-tracking/faster-rcnn.pytorch/lib/model/faster_rcnn/faster_rcnn.py", line 46, in forward
base_feat = self.RCNN_base(im_data)
File "/home/ld/.conda/envs/dave/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/ld/.conda/envs/dave/lib/python3.6/site-packages/torch/nn/modules/container.py", line 91, in forward
input = module(input)
File "/home/ld/.conda/envs/dave/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/ld/.conda/envs/dave/lib/python3.6/site-packages/torch/nn/modules/container.py", line 91, in forward
input = module(input)
File "/home/ld/.conda/envs/dave/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/ld/3d-vehicle-tracking/faster-rcnn.pytorch/lib/model/faster_rcnn/resnet.py", line 86, in forward
out = self.conv1(x)
File "/home/ld/.conda/envs/dave/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/ld/.conda/envs/dave/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 301, in forward
self.padding, self.dilation, self.groups)
RuntimeError: CUDA error: out of memory
The text was updated successfully, but these errors were encountered: