I am trying to implement VGG16, but got below error
Traceback (most recent call last):
File "tools/train_net.py", line 177, in <module>
main()
File "tools/train_net.py", line 170, in main
model = train(cfg, args.local_rank, args.distributed)
File "tools/train_net.py", line 78, in train
arguments,
File "/var/autofs/home/casa/usrs/nagayosi/TextDet/RRPN_pytorch/maskrcnn_benchmark/engine/trainer.py", line 71, in do_train
loss_dict = model(images, targets)
File "/home/usrs/nagayosi/dev/anaconda3/envs/rrpn_pytorch/lib/python3.7/site-packages/torch/nn/modules/module.py", line 489, in __call__
result = self.forward(*input, **kwargs)
File "/var/autofs/home/casa/usrs/nagayosi/TextDet/RRPN_pytorch/maskrcnn_benchmark/modeling/detector/generalized_rrpn_rcnn.py", line 61, in forward
x, result, detector_losses = self.roi_heads(features, proposals, targets)
File "/home/usrs/nagayosi/dev/anaconda3/envs/rrpn_pytorch/lib/python3.7/site-packages/torch/nn/modules/module.py", line 489, in __call__
result = self.forward(*input, **kwargs)
File "/var/autofs/home/casa/usrs/nagayosi/TextDet/RRPN_pytorch/maskrcnn_benchmark/modeling/roi_heads/rroi_heads.py", line 27, in forward
x, detections, loss_box = self.box(features, proposals, targets)
File "/home/usrs/nagayosi/dev/anaconda3/envs/rrpn_pytorch/lib/python3.7/site-packages/torch/nn/modules/module.py", line 489, in __call__
result = self.forward(*input, **kwargs)
File "/var/autofs/home/casa/usrs/nagayosi/TextDet/RRPN_pytorch/maskrcnn_benchmark/modeling/roi_heads/rbox_head/box_head.py", line 57, in forward
class_logits, box_regression = self.predictor(x)
File "/home/usrs/nagayosi/dev/anaconda3/envs/rrpn_pytorch/lib/python3.7/site-packages/torch/nn/modules/module.py", line 489, in __call__
result = self.forward(*input, **kwargs)
File "/var/autofs/home/casa/usrs/nagayosi/TextDet/RRPN_pytorch/maskrcnn_benchmark/modeling/roi_heads/rbox_head/roi_box_predictors.py", line 41, in forward
cls_logit = self.cls_score(x)
File "/home/usrs/nagayosi/dev/anaconda3/envs/rrpn_pytorch/lib/python3.7/site-packages/torch/nn/modules/module.py", line 489, in __call__
result = self.forward(*input, **kwargs)
File "/home/usrs/nagayosi/dev/anaconda3/envs/rrpn_pytorch/lib/python3.7/site-packages/torch/nn/modules/linear.py", line 67, in forward
return F.linear(input, self.weight, self.bias)
File "/home/usrs/nagayosi/dev/anaconda3/envs/rrpn_pytorch/lib/python3.7/site-packages/torch/nn/functional.py", line 1352, in linear
ret = torch.addmm(torch.jit._unwrap_optional(bias), input, weight.t())
RuntimeError: cublas runtime error : resource allocation failed at /pytorch/aten/src/THC/THCGeneral.cpp:250
I implemented VGG16 backbone like below based on resnet.py
Sorry,,,, I solved this issue.
I added new code to modeling/roi_heads/rbox_head/roi_box_feature_extractors.py for vgg implementation, and that code was wrong...
Thank you for your great achievement!
I am trying to implement VGG16, but got below error
I implemented VGG16 backbone like below based on resnet.py
I changed maskrcnn_benchmark/modling/roi_heads/rbox_head/roi_box_predictors.py to below.
my error is occured in line 41, but I have no idea Please help!!