Sayak Paul 9d99489f2f Add TFData2VecVision for semantic segmentation (#17271) ... * feat: initial implementation of data2vec segmentation model in TF. * chore: minor corrections to make the segmenter work. * chore: removed unncessary files. * chore: add tests and other modifications. * fix: loss computation for segmentation. * chore: remove unused variable. * chore: formatting. * added a dummy adaptive pooling layer. * removed unnecessary file. * potentially add identifiers to layer names. * fix: layer naming. * chore: removed unnecessary print. * Skipping unneeded test * chore: add logging to debug tolerance. * fix: segmentation tests for tfdata2vecvision * chore: make style. * fix: layer names, assertion to be resolved. * Bumping test tolerance a bit * chore: bump the tol in PT test. Co-authored-by: matt <rocketknight1@gmail.com>		2022-06-08 14:03:18 +01:00
..
__init__.py	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00
test_modeling_data2vec_audio.py	Black preview (#17217)	2022-05-12 16:25:55 -04:00
test_modeling_data2vec_text.py	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00
test_modeling_data2vec_vision.py	Add TFData2VecVision for semantic segmentation (#17271)	2022-06-08 14:03:18 +01:00
test_modeling_tf_data2vec_vision.py	Add TFData2VecVision for semantic segmentation (#17271)	2022-06-08 14:03:18 +01:00