run for in22k
This commit is contained in:
@@ -1,10 +1,10 @@
|
|||||||
meta:
|
meta:
|
||||||
model_name: vit_huge
|
model_name: vit_huge
|
||||||
tag: in1k
|
tag: in22k
|
||||||
embed_dim: 1280
|
embed_dim: 1280
|
||||||
load_checkpoint: true
|
load_checkpoint: true
|
||||||
checkpoint_folder: experiment_logs/imagenet-vith16.448/
|
checkpoint_folder: experiment_logs/imagenet-vith16.448/
|
||||||
read_checkpoint: IN1K-vit.h.14-300e.pth.tar
|
read_checkpoint: IN22K-vit.h.14-900e.pth.tar
|
||||||
use_bfloat16: true
|
use_bfloat16: true
|
||||||
num_classes: 182
|
num_classes: 182
|
||||||
representation_type: last_avgpool
|
representation_type: last_avgpool
|
||||||
|
|||||||
Reference in New Issue
Block a user