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