try to set steps in predict function

model.predict(test_dataset_sim, steps=steps)

and the steps is the length of your test_dataset_sim / test_batch_sz

thanks.