SforAiDl / vformer

A modular PyTorch library for vision transformer models
https://vformer.readthedocs.io/
MIT License
162 stars 22 forks source link

Add ViViT implementation #66

Closed abhi-glitchhg closed 2 years ago

abhi-glitchhg commented 2 years ago

Here is a model3 of ViViT

cc @NeelayS @aditya-agrawal-30502

codecov-commenter commented 2 years ago

Codecov Report

Merging #66 (015b10b) into main (36102f3) will not change coverage. The diff coverage is 100.00%.

@@            Coverage Diff            @@
##              main       #66   +/-   ##
=========================================
  Coverage   100.00%   100.00%           
=========================================
  Files           61        62    +1     
  Lines         1749      1804   +55     
=========================================
+ Hits          1749      1804   +55     
Impacted Files Coverage Δ
vformer/encoder/__init__.py 100.00% <100.00%> (ø)
vformer/encoder/vivit.py 100.00% <100.00%> (ø)
vformer/models/classification/__init__.py 100.00% <100.00%> (ø)
vformer/models/classification/vivit.py 100.00% <100.00%> (ø)