the weights files not works

[xiaofengShi]

I changed the attention.py # out,state = tflib.ops.FreeRunIm2LatexAttention('AttLSTM', ctx, emb_seqs, EMB_DIM, ENC_DIM, DEC_DIM, D, H, W) to out, state = tflib.ops.im2latexAttention('AttLSTM', emb_seqs, ctx, EMB_DIM, ENC_DIM, DEC_DIM, D, H, W) ,the net runs correctly but the weight files has an error NotFoundError (see above for traceback): Key rnn/while/AttLSTM.AttentionCell.Gates/AttLSTM.AttentionCell.Gates.W not found in checkpoint [[Node: save/RestoreV2 = RestoreV2[dtypes=[DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, ..., DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT], _device="/job:localhost/replica:0/task:0/device:CPU:0"](_arg_save/Const_0_0, save/RestoreV2/tensor_names, save/RestoreV2/shape_and_slices)]]

[qianwangn]

use predict.py