Issue with batch normalization layer of saved CNN

When loading a previously trained CNN, I got an issue with the batch normalization layers. When looking into the loaded CNN model the trainable mean and variance are empty.

Name: ‘batchnorm_1’

TrainedMean: []

TrainedVariance: []

So the checkpoint doesn’t seem to save these parameters. Are there any workarounds for this issue? I am using Matlab R2018b.

ANSWER

Matlabsolutions.com provide latest MatLab Homework Help,MatLab Assignment Help for students, engineers and researchers in Multiple Branches like ECE, EEE, CSE, Mechanical, Civil with 100% output.Matlab Code for B.E, B.Tech,M.E,M.Tech, Ph.D. Scholars with 100% privacy guaranteed. Get MATLAB projects with source code for your learning and research.

We were able to reproduce the issue. We will inform you once the issue is fixed.

Since TrainedMean and TrainedVariance are calculated after the training is finished, therefore as a workaround you can use the below mentioned codes to explicitly save and load the Model.

%To save model with name "demoModel", assuming your network is in "net" 
save('demoModel','net')%To load model to variable net1
net1=load('demoModel.mat','net');
net1.net.Layers(n).TrainedMean %where n is the batch normalization layer

SEE COMPLETE ANSWER CLICK THE LINK