Fixing parameters in pretrained model?


#1

Hi, I’m fairly new to mxnet and am trying to do some fine-tuning of ResNet for my particular project. I’d like to fix the parameters in earlier layers in ResNet so that I only update the later layers - how can I do this with the model returned by from mxnet.gluon.model_zoo.vision.models.resnet(whatever version)?


#2

You can get the model for Resnet from here: https://github.com/apache/incubator-mxnet/blob/master/python/mxnet/gluon/model_zoo/vision/resnet.py#L230 . You can reuse this code and customize it to tweak the parameters.