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


model.embed_dim = 384
model.depth = 12
model.num_heads = 12
model.drop_path_rate = 0.0