vc1-base / config.yaml
sneha
fix
4da3be6
raw
history blame contribute delete
431 Bytes
_target_: vc_models.models.load_model
model:
_target_: vc_models.models.vit.vit.load_mae_encoder
checkpoint_path: ./vc1_vitb.pth
model:
_target_: vc_models.models.vit.vit.vit_base_patch16
img_size: 224
use_cls: True
drop_path_rate: 0.0
transform:
_target_: vc_models.transforms.vit_transforms
metadata:
algo: mae
model: vit_base_patch16
data:
- ego
- imagenet
- inav
comment: 182_epochs