How to control the end of running on test dataset?

google-research / tapas

End-to-end neural table-text understanding models.

Apache License 2.0

1.15k stars 217 forks source link

How to control the end of running on test dataset? #57

Closed lairikeqiA closed 4 years ago

lairikeqiA commented 4 years ago

When the mode is "predict and evaluate", the model always repeats prediction and evaluation. How do I control the model to stop after running a time of prediction and evaluation?

muelletm commented 4 years ago

You should be able to disable that behavior by setting 'loop_predict' to false.

lairikeqiA commented 4 years ago

Thanks!