vit_large_patch16.py 99 Bytes
Newer Older
yuguo960516's avatar
yuguo960516 committed
1
2
3
4
5
6
from .vit_base_patch16 import model


model.embed_dim = 1024
model.depth = 24
model.num_heads = 16