sneha
commited on
Commit
•
19bd325
1
Parent(s):
acd5de4
change dict structure of model weights, rename library in config to vc_models
Browse files- config.yaml +5 -5
- pytorch_model.bin +2 -2
config.yaml
CHANGED
@@ -1,14 +1,14 @@
|
|
1 |
-
_target_:
|
2 |
model:
|
3 |
-
_target_:
|
4 |
-
checkpoint_path: ./
|
5 |
model:
|
6 |
-
_target_:
|
7 |
img_size: 224
|
8 |
use_cls: True
|
9 |
drop_path_rate: 0.0
|
10 |
transform:
|
11 |
-
_target_:
|
12 |
metadata:
|
13 |
algo: mae
|
14 |
model: vit_base_patch16
|
|
|
1 |
+
_target_: vc_models.load_model
|
2 |
model:
|
3 |
+
_target_: vc_models.models.vit.vit.load_mae_encoder
|
4 |
+
checkpoint_path: ./vc1_vitb.pth
|
5 |
model:
|
6 |
+
_target_: vc_models.models.vit.vit.vit_base_patch16
|
7 |
img_size: 224
|
8 |
use_cls: True
|
9 |
drop_path_rate: 0.0
|
10 |
transform:
|
11 |
+
_target_: vc_models.transforms.vit_transforms
|
12 |
metadata:
|
13 |
algo: mae
|
14 |
model: vit_base_patch16
|
pytorch_model.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1321150ee7e45c5c1a8ea2bb6181cce87841f87f94ec69f0a451cfe4a63b77ba
|
3 |
+
size 447726469
|