Sayak Paul 5b40a37bc4 Add TF ViT MAE (#16255) ... * ported TFViTMAEIntermediate and TFViTMAEOutput. * added TFViTMAEModel and TFViTMAEDecoder. * feat: added a noise argument in the implementation for reproducibility. * feat: vit mae models with an additional noise argument for reproducibility. Co-authored-by: ariG23498 <aritra.born2fly@gmail.com> Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>		2022-03-29 18:24:15 +01:00
..
__init__.py	[Test refactor 1/5] Per-folder tests reorganization (#15725)	2022-02-23 15:46:28 -05:00
test_modeling_tf_vit_mae.py	Add TF ViT MAE (#16255)	2022-03-29 18:24:15 +01:00
test_modeling_vit_mae.py	Add TF ViT MAE (#16255)	2022-03-29 18:24:15 +01:00