Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

How to change layers in pretrained model in Mxnet

Tags:

python

mxnet

I have a pre-trained Mxnet model. I need to change last two layers and add new two layers for testing. Basically, I need to create a probability map of the image. How can i do that ?

like image 582
Ashwin Raju Avatar asked Oct 18 '22 05:10

Ashwin Raju


1 Answers

There is a tutorial on fine-tuning with MXNet. Did you check this out? http://mxnet.incubator.apache.org/faq/finetune.html

like image 154
eric-haibin-lin Avatar answered Oct 22 '22 10:10

eric-haibin-lin