faster_rcnn_pytorch icon indicating copy to clipboard operation
faster_rcnn_pytorch copied to clipboard

out of memory if don`t fix VGG16 param

Open aRookieMan opened this issue 7 years ago • 1 comments

If dont fix pretrained VGG16s parameters , the model will increasing untill memory booms. But if fix these parameters , it works. Why?

code in train.py

for param in net.rpn.features.parameters():
    param.requires_grad = False

aRookieMan avatar Dec 07 '18 01:12 aRookieMan

谢谢

SmallSmallQiu avatar Sep 27 '19 13:09 SmallSmallQiu