RuntimeError: mae_label is not presented, while testing?

I have trained a ResNet-100 model on my dataset whose output is a float (the Last layer is MAERegressionOutput().

The script is as below:
import mxnet as mx
import argparse
import pickle
import os
import numpy as np

testIter =
	data_shape=(3, 112, 112),

epoch = 4
(symbol, argParams, auxParams) = mx.model.load_checkpoint('age-mse-tutorial-model-r100s', epoch)

# construct the model
model = mx.mod.Module(symbol=symbol, context=[mx.gpu(0)])
model.bind(data_shapes=[('data', (1, 3, 112, 112))], 
           label_shapes=[('mae_label', (1,))] )
model.set_params(argParams, auxParams)

metric = mx.metric.MAE()
score = model.score(testIter, metric)

The error which is I am getting is

[13:25:28] src/io/ ImageRecordIOParser2: test.rec, use 4 threads for decoding..
/home/anubhav/anaconda3/envs/trueface3.6/lib/python3.6/site-packages/mxnet/module/ UserWarning: You created Module with Module(..., label_names=['softmax_label']) but input with name 'softmax_label' is not found in symbol.list_arguments(). Did you mean one of:
/home/anubhav/anaconda3/envs/trueface3.6/lib/python3.6/site-packages/mxnet/module/ UserWarning: Data provided by label_shapes don't match names specified by label_names ([DataDesc[mae_label,(1,),<class 'numpy.float32'>,NCHW]] vs. ['softmax_label'])
Traceback (most recent call last):
  File "", line 27, in <module>
    model.set_params(argParams, auxParams)
  File "/home/anubhav/anaconda3/envs/trueface3.6/lib/python3.6/site-packages/mxnet/module/", line 350, in set_params
  File "/home/anubhav/anaconda3/envs/trueface3.6/lib/python3.6/site-packages/mxnet/module/", line 309, in init_params
    _impl(desc, arr, arg_params)
  File "/home/anubhav/anaconda3/envs/trueface3.6/lib/python3.6/site-packages/mxnet/module/", line 300, in _impl
    raise RuntimeError("%s is not presented" % name)
RuntimeError: mae_label is not presented

Could anyone suggest what’s wrong here?

This looks like a MXNet framework that is throwing error. Could you tag this as MXNet?