ashkamath / mdetr

Apache License 2.0
969 stars 125 forks source link

Freeze transformers package version #23

Closed nguyeho7 closed 3 years ago

nguyeho7 commented 3 years ago

Latest transformers package (4.9.1) breaks the ROBERTA text encoder during pretraining, so we freeze it at version 4.5.1

alcinos commented 3 years ago

Thanks for your contribution @nguyeho7! Merging this.