tests/test_modeling_tf_common.py · 6589e510fa4e6c442059de2fab84752535de9b23 · chenpangpang / transformers

Use shape_list to safely get shapes for Swin (#17591) · 9fc34235

amyeroberts authored Jun 09, 2022

* Use shape_list to safely get shapes

* Add relevant test

* Tidy and add metrics

* Resolve dynamic shaping issues and move test

* Tidy up and all samples in batch

* Formatting

9fc34235

test_modeling_tf_common.py 88.7 KB

Replace test_modeling_tf_common.py