gaunernst commited on
Commit
1d0bc2d
1 Parent(s): be1ccb8

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +21 -0
config.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architecture": "convnext_atto",
3
+ "num_classes": 512,
4
+ "num_features": 320,
5
+ "pretrained_cfg": {
6
+ "input_size": [3, 112, 112],
7
+ "fixed_input_size": false,
8
+ "interpolation": "bicubic",
9
+ "crop_pct": 1.0,
10
+ "crop_mode": "center",
11
+ "mean": [0.5, 0.5, 0.5],
12
+ "std": [0.5, 0.5, 0.5],
13
+ "num_classes": 512,
14
+ "pool_size": [7, 7],
15
+ "first_conv": "stem.0",
16
+ "classifier": "head.fc"
17
+ },
18
+ "model_args": {
19
+ "patch_size": 2
20
+ }
21
+ }