error: UnboundLocalError: local variable 'mask_batch' referenced before assignment

[lizhiaizhanrui]

Hello,Thank you very much for you work,I am student with no experience in deep learning and image inpainting. i download your code and try to run it . but when i train the mode by using python train.py, i got a error:
Traceback (most recent call last):
File "train.py", line 138, in
train()
File "train.py", line 66, in train
completion = sess.run(model.completion, feed_dict={x: x_batch, mask: mask_batch, is_training: False})
UnboundLocalError: local variable 'mask_batch' referenced before assignment

I saw the source code and try to solve it, but didn't work at all. if you know why,would you please tell me? thanks very much.