DineshKumar1329 commited on
Commit
ccf7fd6
·
verified ·
1 Parent(s): ab6d81c

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +8 -0
config.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "ResNet18",
3
+ "input_size": [3, 128, 128],
4
+ "output_size": 2,
5
+ "class_names": ["Cat", "Dog"],
6
+ "mean": [0.485, 0.456, 0.406],
7
+ "std": [0.229, 0.224, 0.225]
8
+ }