Phil Wang 22da26fa4b fix recorder in data parallel situation

2021-07-08 10:15:07 -07:00

..

__init__.py

implement SOTA new self-supervised learning technique from facebook for vision transformers, Dino

2021-05-02 14:00:36 -07:00

cait.py

add layer dropout for CaiT

2021-04-01 20:30:37 -07:00

cct.py

Minor changes

2021-07-01 17:51:35 -07:00

cross_vit.py

add documentation for three recent vision transformer follow-up papers

2021-03-31 09:22:15 -07:00

cvt.py

fix norm issues in cvt

2021-04-27 08:36:17 -07:00

deepvit.py

add deep vit

2021-03-23 11:57:13 -07:00

dino.py

implement SOTA new self-supervised learning technique from facebook for vision transformers, Dino

2021-05-02 14:00:36 -07:00

distill.py

Fix alpha coefficient multiplication in the loss

2021-04-14 11:36:43 +02:00

efficient.py

make sure distillation still works

2021-02-21 19:08:18 -08:00

levit.py

apply scale to 2d rel pos bias in levit

2021-05-10 11:37:23 -07:00

local_vit.py

fix pooling bugs across a few new archs

2021-04-19 22:36:23 -07:00

mpp.py

fix mpp

2021-06-16 16:46:32 -07:00

nest.py

layernorm on channel dimension == instancenorm2d with affine set to true

2021-06-03 16:41:45 -07:00

pit.py

for PiT, project to increased dimensions on first grouped conv for depthwise-conv

2021-04-29 12:41:00 -07:00

recorder.py

fix recorder in data parallel situation

2021-07-08 10:15:07 -07:00

rvt.py

Add use_glu flag to RvT

2021-04-30 02:07:41 -04:00

t2t.py

remove masking, as it complicates with little benefit

2021-03-23 12:18:47 -07:00

twins_svt.py

fix small bug

2021-04-29 15:39:41 -07:00

vit.py

be able to accept non-square patches, thanks to @FilipAndersson245

2021-05-01 20:04:41 -07:00