sneha
commited on
Commit
·
68fefc5
1
Parent(s):
57dd881
mapping key error
Browse files- config.yaml +1 -1
config.yaml
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
_target_: eaif_models.models.load_model
|
| 2 |
model:
|
| 3 |
_target_: eaif_models.models.vit.vit.load_mae_encoder
|
| 4 |
-
checkpoint_path:
|
| 5 |
model:
|
| 6 |
_target_: eaif_models.models.vit.vit.vit_base_patch16
|
| 7 |
img_size: 224
|
|
|
|
| 1 |
_target_: eaif_models.models.load_model
|
| 2 |
model:
|
| 3 |
_target_: eaif_models.models.vit.vit.load_mae_encoder
|
| 4 |
+
checkpoint_path: ./pytorch_model.bin
|
| 5 |
model:
|
| 6 |
_target_: eaif_models.models.vit.vit.vit_base_patch16
|
| 7 |
img_size: 224
|