如何使用pytorch删除Faster-RCNN中的图层

Question

如何删除Faster RCNN中的fc7层？ 以下是模型结构的一部分：

(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内存将立即从7GB变为15GB。 这里有什么问题？