sneha commited on
Commit
19bd325
1 Parent(s): acd5de4

change dict structure of model weights, rename library in config to vc_models

Browse files
Files changed (2) hide show
  1. config.yaml +5 -5
  2. pytorch_model.bin +2 -2
config.yaml CHANGED
@@ -1,14 +1,14 @@
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
8
  use_cls: True
9
  drop_path_rate: 0.0
10
  transform:
11
- _target_: eaif_models.transforms.vit_transforms
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:85779bc30e012ece9c742605327ce7f598c0d2f3615e03c15bc43d76f453b8d8
3
- size 447726405
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1321150ee7e45c5c1a8ea2bb6181cce87841f87f94ec69f0a451cfe4a63b77ba
3
+ size 447726469