ash001 commited on
Commit
150b99e
·
verified ·
1 Parent(s): 249f22a

Upload folder using huggingface_hub

Browse files
artifacts/class_names.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ [
2
+ "cats",
3
+ "dogs"
4
+ ]
artifacts/config.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "resnet18",
3
+ "num_classes": 2,
4
+ "image_size": 224,
5
+ "mean": [
6
+ 0.485,
7
+ 0.456,
8
+ 0.406
9
+ ],
10
+ "std": [
11
+ 0.229,
12
+ 0.224,
13
+ 0.225
14
+ ],
15
+ "notes": "Transfer learning: ResNet-18 backbone frozen, trained final FC layer."
16
+ }
artifacts/model_state.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7080f54b5ace1a8005567c75a0b62a764b46d50e03114a43a9bcc5e7746e5757
3
+ size 44789899