Как удалить слои в Faster-RCNN с помощью pytorch - PullRequest
Новая
       46

Как удалить слои в Faster-RCNN с помощью pytorch

0 голосов
/ 02 мая 2020

Как я могу удалить слой fc7 в Faster RCNN? И следующее является частью структуры модели: 

(roi_heads): RoIHeads(
(box_roi_pool): MultiScaleRoIAlign()
(box_head): TwoMLPHead(
  (fc6): Linear(in_features=12544, out_features=1024, bias=True)
  (fc7): Linear(in_features=1024, out_features=1024, bias=True)
)
(box_predictor): FastRCNNPredictor(
  (cls_score): Linear(in_features=1024, out_features=91, bias=True)
  (bbox_pred): Linear(in_features=1024, out_features=364, bias=True)
)

И если я это сделаю: 

model.roi_heads.box_head=model.roi_heads.box_head.fc6

Размер модели станет супер огромным. Когда я помещаю модель в cuda, память CUDA сразу же увеличит go с 7 до 15 ГБ. В чем здесь проблема?

...