Add missing flags for pretrain experiment (AttributeError)

google-research / tapas

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

Apache License 2.0

1.15k stars 217 forks source link

Add missing flags for pretrain experiment (AttributeError) #189

Open Maxvgrad opened 4 months ago

Maxvgrad commented 4 months ago

Fixing error an AttributeError for the restrict_attention_bucket_size flag is raised when running the tapas_pretraining_experiment script.

Explicitly providing the command line flag doesn't solve the the problem (Unknown command line flag 'restrict_attention_bucket_size'). Probably, the flag was not defined in the script file during changes related to MATE

Resolve the issue: https://github.com/google-research/tapas/issues/184

Maxvgrad commented 4 months ago

Hi @eisenjulian, could you please have a look? 🙇