wucng commited on
Commit
07255f5
1 Parent(s): 2bf9f63

Upload model

Browse files
Files changed (2) hide show
  1. config.json +1 -1
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_name_or_path": "custom-tvres18",
3
  "architectures": [
4
- "ResnetModel"
5
  ],
6
  "input_channels": 3,
7
  "model_name": "resnet18",
 
1
  {
2
  "_name_or_path": "custom-tvres18",
3
  "architectures": [
4
+ "ResnetModelForImageClassification"
5
  ],
6
  "input_channels": 3,
7
  "model_name": "resnet18",
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b072267afe3d6da983e3e98aeb7222257736d592046d442a42197e223d6b6fc6
3
- size 44755968
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bff9e5fa379b6f6a5da5c2f838bf4f8dc91e8f17b5fc65373b5d23efeae73d3b
3
+ size 46808208