[NQG] AssertionError: Found 392 Python objects that were not bound to checkpointed values

Hi,

Thanks for sharing the code. I was trying to replicate the NQG results on GeoQuery (https://github.com/google-research/language/tree/master/language/compgen/nqg). After finished up with QCFG induction step and generating training data steps, I downloaded BERT-uncased model checkpoint from this link: https://github.com/tensorflow/models/blob/master/official/nlp/docs/pretrained_models.md because the original link in repository is broken.

I used the command below for training.
```
python model/parser/training/train_model.py \
--input=${TF_EXAMPLES} \
--config=${CONFIG} \
--model_dir=${MODEL_DIR} \
--bert_dir=${BERT_DIR}
```

However, I am receiving error below, and the training for NQG could not be proceeded. Could you please point me to the exact model checkpoint that you were using in NQG and what could be the reason here?

```
Traceback (most recent call last):
File "/private/home/kaisersun/CompGenComparision/baseline_replication/TMCD/model/parser/training/train_model.py", line 139, in
app.run(main)
File "/private/home/kaisersun/.conda/envs/gen/lib/python3.9/site-packages/absl/app.py", line 308, in run
_run_main(main, args)
File "/private/home/kaisersun/.conda/envs/gen/lib/python3.9/site-packages/absl/app.py", line 254, in _run_main
sys.exit(main(argv))
File "/private/home/kaisersun/CompGenComparision/baseline_replication/TMCD/model/parser/training/train_model.py", line 135, in main
train_model(strategy)
File "/private/home/kaisersun/CompGenComparision/baseline_replication/TMCD/model/parser/training/train_model.py", line 99, in train_model
status.assert_existing_objects_matched()
File "/private/home/kaisersun/.conda/envs/gen/lib/python3.9/site-packages/tensorflow/python/training/tracking/util.py", line 846, in assert_existing_objects_matched
raise AssertionError(
AssertionError: Found 392 Python objects that were not bound to checkpointed values, likely due to changes in the Python program. Showing 10 of 392 unmatched objects: [, , , , , , , , , ]
```