MHanzl commited on
Commit
8c2db7c
1 Parent(s): 0d25ff0

Fix config

Browse files
Files changed (1) hide show
  1. config.json +7 -2
config.json CHANGED
@@ -1,9 +1,14 @@
1
  {
2
  "architecture": "vit_large_patch16_384",
3
- "input_size": [
 
 
4
  3,
5
  384,
6
  384
7
- ],
 
 
 
8
  "num_classes": 182
9
  }
 
1
  {
2
  "architecture": "vit_large_patch16_384",
3
+ "pretrained_cfg": {
4
+ "custom_load": false,
5
+ "input_size": [
6
  3,
7
  384,
8
  384
9
+ ],
10
+ "fixed_input_size": true,
11
+ "first_conv": "patch_embed.proj"
12
+ },
13
  "num_classes": 182
14
  }