Dricz commited on
Commit
0b5537a
·
verified ·
1 Parent(s): 453add9

Training in progress, epoch 1

Browse files
config.json CHANGED
@@ -1,32 +1,49 @@
1
  {
2
- "_name_or_path": "google/vit-base-patch16-224-in21k",
3
  "architectures": [
4
- "ViTForImageClassification"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
  ],
6
- "attention_probs_dropout_prob": 0.0,
7
- "encoder_stride": 16,
8
- "hidden_act": "gelu",
9
- "hidden_dropout_prob": 0.0,
10
- "hidden_size": 768,
11
  "id2label": {
12
  "0": "cat",
13
  "1": "dog"
14
  },
15
- "image_size": 224,
16
- "initializer_range": 0.02,
17
- "intermediate_size": 3072,
18
  "label2id": {
19
  "cat": "0",
20
  "dog": "1"
21
  },
22
- "layer_norm_eps": 1e-12,
23
- "model_type": "vit",
24
- "num_attention_heads": 12,
25
  "num_channels": 3,
26
- "num_hidden_layers": 12,
27
- "patch_size": 16,
 
 
 
 
28
  "problem_type": "single_label_classification",
29
- "qkv_bias": true,
 
 
 
 
 
 
30
  "torch_dtype": "float32",
31
  "transformers_version": "4.37.2"
32
  }
 
1
  {
2
+ "_name_or_path": "microsoft/resnet-50",
3
  "architectures": [
4
+ "ResNetForImageClassification"
5
+ ],
6
+ "depths": [
7
+ 3,
8
+ 4,
9
+ 6,
10
+ 3
11
+ ],
12
+ "downsample_in_bottleneck": false,
13
+ "downsample_in_first_stage": false,
14
+ "embedding_size": 64,
15
+ "hidden_act": "relu",
16
+ "hidden_sizes": [
17
+ 256,
18
+ 512,
19
+ 1024,
20
+ 2048
21
  ],
 
 
 
 
 
22
  "id2label": {
23
  "0": "cat",
24
  "1": "dog"
25
  },
 
 
 
26
  "label2id": {
27
  "cat": "0",
28
  "dog": "1"
29
  },
30
+ "layer_type": "bottleneck",
31
+ "model_type": "resnet",
 
32
  "num_channels": 3,
33
+ "out_features": [
34
+ "stage4"
35
+ ],
36
+ "out_indices": [
37
+ 4
38
+ ],
39
  "problem_type": "single_label_classification",
40
+ "stage_names": [
41
+ "stem",
42
+ "stage1",
43
+ "stage2",
44
+ "stage3",
45
+ "stage4"
46
+ ],
47
  "torch_dtype": "float32",
48
  "transformers_version": "4.37.2"
49
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d7869dc01f19f2fb2227b8d4b82739a4f8505569ded7abe70ccd2a2d1297d7b3
3
- size 343223968
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dac7136d5bd1e495149637ae33ec14d8295be3f5805102edcc1a6c7914613e24
3
+ size 94302952
preprocessor_config.json CHANGED
@@ -1,22 +1,22 @@
1
  {
 
2
  "do_normalize": true,
3
  "do_rescale": true,
4
  "do_resize": true,
5
  "image_mean": [
6
- 0.5,
7
- 0.5,
8
- 0.5
9
  ],
10
- "image_processor_type": "ViTImageProcessor",
11
  "image_std": [
12
- 0.5,
13
- 0.5,
14
- 0.5
15
  ],
16
- "resample": 2,
17
  "rescale_factor": 0.00392156862745098,
18
  "size": {
19
- "height": 224,
20
- "width": 224
21
  }
22
  }
 
1
  {
2
+ "crop_pct": 0.875,
3
  "do_normalize": true,
4
  "do_rescale": true,
5
  "do_resize": true,
6
  "image_mean": [
7
+ 0.485,
8
+ 0.456,
9
+ 0.406
10
  ],
11
+ "image_processor_type": "ConvNextImageProcessor",
12
  "image_std": [
13
+ 0.229,
14
+ 0.224,
15
+ 0.225
16
  ],
17
+ "resample": 3,
18
  "rescale_factor": 0.00392156862745098,
19
  "size": {
20
+ "shortest_edge": 224
 
21
  }
22
  }
runs/Feb23_10-48-12_49e4c2c8e75d/events.out.tfevents.1708685293.49e4c2c8e75d.1841.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5596eaa3e7e5001f332cf812892a70a3650ca66df544775a52f3e880b63b648d
3
+ size 5445
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:16a1a393667edcdefd577268a8d9803ec5c9124b523badbc6017e4743e7979d4
3
  size 4728
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7713e54fae664eb01b73875a6490ded5fba009929442a98e6f56c433eef007ce
3
  size 4728