finetune with my data

[lxltaibo]

Hi, I want to use the IEF to train with my own data, how can do it?

I try to input the data like this:
input: "image"
input_dim: 1
input_dim: 3
input_dim: 224
input_dim: 224
input: "kp_pos"
input_dim: 1
input_dim: 17
input_dim: 2
input_dim: 1
input: "label"
input_dim: 1
input_dim: 16
input_dim: 2
input_dim: 1

but I got the error as:
I0221 09:49:03.289826 27112 caffe.cpp:219] Starting Optimization
I0221 09:49:03.289835 27112 solver.cpp:280] Solving
I0221 09:49:03.289839 27112 solver.cpp:281] Learning Rate Policy: poly
Using Config:
Namespace(K=100.0, T=-50.0, imgSz=224, sigma=0.001)
Using Config:
Namespace(K=100.0, T=-50.0, imgSz=224, sigma=0.001)
Traceback (most recent call last):
File "/path../caffe_ief/python_layer/src/layers/python_ief.py", line 71, in forward
top[0].data[b][k,yImSt:yImEn,xImSt:xImEn] = copy.deepcopy(self.g_[ySt:yEn, xSt:xEn])
ValueError: could not broadcast input array from shape (410,224) into shape (0,224)