zhaoli2017

Yes, I changed the BASE_PRETRAINED_DIR. I can fine-tune and evaluate the model, however, there is an error when running 'model.predict'. `/usr/local/lib/python3.7/dist-packages/mesh_tensorflow/transformer/utils.py in infer_model(estimator, vocabulary, sequence_length, batch_size, model_type, model_dir, eval_checkpoint_step, checkpoint_paths,...