fail to froze the graph

[sillytheif]

Hi, I want to froze the model to conduct unittest. When I run the command "g2p-seq2seq --model_dir model_folder_bre --freeze" There exisits the bug: AssertionError: transformer/parallel_0_5/transformer/body/decoder/layer_0/self_attention/multihead_attention/dot_product_attention/Softmax is not in graph. I don't know how to fix this bug. I'll will be grateful for your help.

[entenbein]

Same here, hints would be appreciated!