KingMan1980 commited on
Commit
6289697
·
verified ·
1 Parent(s): fc2d959

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": "brown_box",
13
  "1": "white_box"
14
  },
15
- "image_size": 224,
16
- "initializer_range": 0.02,
17
- "intermediate_size": 3072,
18
  "label2id": {
19
  "brown_box": "0",
20
  "white_box": "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.49.0"
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": "brown_box",
24
  "1": "white_box"
25
  },
 
 
 
26
  "label2id": {
27
  "brown_box": "0",
28
  "white_box": "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.49.0"
49
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e0241b025bf61f3f54f17b3f791b1ae405bbc625e9fea9c00d801a3536748f6a
3
- size 343223968
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7c8e39277381fdf6022ea8b87646eac291c83012316c79ca9687066a0cf38166
3
+ size 94302952
preprocessor_config.json CHANGED
@@ -1,23 +1,22 @@
1
  {
2
- "do_convert_rgb": null,
3
  "do_normalize": true,
4
  "do_rescale": true,
5
  "do_resize": true,
6
  "image_mean": [
7
- 0.5,
8
- 0.5,
9
- 0.5
10
  ],
11
- "image_processor_type": "ViTImageProcessor",
12
  "image_std": [
13
- 0.5,
14
- 0.5,
15
- 0.5
16
  ],
17
- "resample": 2,
18
  "rescale_factor": 0.00392156862745098,
19
  "size": {
20
- "height": 224,
21
- "width": 224
22
  }
23
  }
 
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/Feb20_13-06-35_Mengs-MacBook-Pro.local/events.out.tfevents.1740027996.Mengs-MacBook-Pro.local.15907.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:27546f37a8704faefd832c2d09ef141936978cf8d9ac752e165035998a8914aa
3
- size 22252
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e280ff2268f534168613af503382e5b5d23be5293462fe40d1b980748b17a239
3
+ size 22674
runs/Feb20_13-29-57_Mengs-MacBook-Pro.local/events.out.tfevents.1740029398.Mengs-MacBook-Pro.local.16857.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c1465aa01da6c6921266baeb721e2bfa12c2e16baa7353ae288d00442dcefda4
3
+ size 6149
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:825fb4c6c165ba451fcba9a2514571f79ddfc93b99f854a1c7b21de088a80bbf
3
  size 5368
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b636e47e1ce1a0b1d18bae1a7d7c56f577c0564c72c06d7f2c67dfa95cd40aae
3
  size 5368