Mode | Name | Size | |
---|---|---|---|
-rw-r--r-- | __init__.py | 91 | logplain |
-rw-r--r-- | attention.py | 4078 | logplain |
-rw-r--r-- | layers.py | 3687 | logplain |
-rw-r--r-- | mlp.py | 1271 | logplain |
-rw-r--r-- | norm.py | 973 | logplain |
d--------- | nystromer | 119 | logplain |
d--------- | positional_encodings | 185 | logplain |
-rw-r--r-- | residual.py | 218 | logplain |
-rw-r--r-- | transformer.py | 1822 | logplain |
-rw-r--r-- | vit.py | 1360 | logplain |