shoaib6174 / GSOC-22-Video-Swin-Transformers

MIT License
3 stars 1 forks source link

Batch Size Extraction in Eager Execution #7

Closed Aman-Verma-2307 closed 11 months ago

Aman-Verma-2307 commented 11 months ago

Hi: Thanks for this really nice implementation!

I had a query: Batch-Size cannot be extracted during eager-execution, however in window_partition and window_reverse you are extracting Batch-Size via tf.shape(), is this not going to throw an error? Alternatively, this code bears compatibility till what version of tensorflow?