pred depth
This commit is contained in:
@@ -35,7 +35,7 @@ meta:
|
||||
load_checkpoint: false
|
||||
model_name: vit_base
|
||||
embed_dim: 768
|
||||
pred_depth: 12
|
||||
pred_depth: 6
|
||||
pred_emb_dim: 384
|
||||
read_checkpoint: null
|
||||
use_bfloat16: true
|
||||
|
||||
Reference in New Issue
Block a user