ShownX opened a new issue #12334: Load pretrained model with new 
load_parameters() API
URL: https://github.com/apache/incubator-mxnet/issues/12334
 
 
   ## Description
   Cannot load only convolutional layers with pre-trained model using 
```load_parameters()``` API 
   
   ## Environment info (Required)
   Ubuntu 16.04
   mxnet-cu90 1.3.0b20180821
   
   ## Error Message:
   missing_parameters
   
   ## Minimum reproducible example
   ```
   from mxnet.gluon.model_zoo.vision import get_model
   
   # download the pretrained model with the path: model_path
   model = get_model('resnet101_v2')
   model.features.load_parameters(model_path)
   ```
   
   ## What have you tried to solve it?
   The reason of this issue is load_parameters called from ```features 
block```, in which name of parameters starts with numbers. But the pretrained 
model parameters starts with ```features```.
   The ```load_parameters``` cannot find the corresponding name in this case.
   
   One way to work around this issue is using the ```model.save_params()``` to 
save again and load using the API ```load_parameters```. However, the 
save_params() is deprecated.
   
   To prevent any problem with loading parameters, I propose this issue and 
hope the team can fix it.

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
[email protected]


With regards,
Apache Git Services

Reply via email to