Mode | Name | Size | |
---|---|---|---|
-rw-r--r-- | __init__.py | 27 | logplain |
-rw-r--r-- | attention.py | 4284 | logplain |
d--------- | axial_attention | 158 | logplain |
d--------- | embeddings | 150 | logplain |
-rw-r--r-- | layers.py | 3086 | logplain |
-rw-r--r-- | mlp.py | 1271 | logplain |
-rw-r--r-- | norm.py | 1141 | logplain |
-rw-r--r-- | residual.py | 245 | logplain |
-rw-r--r-- | vit.py | 1360 | logplain |