mae_vit_large_patch16.py 103 Bytes
Newer Older
yuguo960516's avatar
gpt2  
yuguo960516 committed
1
2
3
4
5
6
from .mae_vit_base_patch16 import model


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