vc1-large / config.yaml
sneha
change eaif_models to vc_models
84d8923
_target_: vc_models.models.load_model
model:
_target_: vc_models.models.vit.vit.load_mae_encoder
checkpoint_path: model_ckpts/vc1_vitl.pth
model:
_target_: vc_models.models.vit.vit.vit_large_patch16
img_size: 224
use_cls: True
drop_path_rate: 0.0
transform:
_target_: vc_models.transforms.vit_transforms
metadata:
algo: mae
model: vit_large_patch16
data:
- ego
- imagenet
- inav
comment: 182_epochs