ThomasDelteil edited a comment on issue #11522: MXNet 1.2.0 - net.save_params() 
output parameter file contains garbage
URL: 
https://github.com/apache/incubator-mxnet/issues/11522#issuecomment-402886569
 
 
   @simonmaurer 
   please use 
   ```python
   net.hybridize()
   net(mx.nd.ones((1,3,224,224))
   net.export('test')
   ```
   To export your model.
   You will get a `test-symbol.json` and `test-0000.params` file.
   
   For more details refer to this tutorial: 
https://mxnet.incubator.apache.org/tutorials/gluon/save_load_params.html
   
   edit: Also note that if you want to fine-tune VGG16 with a different final 
layer, I would suggest this much simpler way:
   
   ```python
   vgg16 = gluon.model_zoo.vision.vgg16(pretrained=True)
   with vgg16.name_scope():
       vgg16.output = mx.gluon.nn.Dense(32, prefix='bit_layer')
   vgg16.output.initialize()
   
   vgg16.hybridize()
   vgg16(mx.nd.ones((1,3,224,224)))
   vgg16.export('test')
   ```
   

----------------------------------------------------------------
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