dima806 commited on
Commit
4664652
1 Parent(s): a68e8fc

Upload folder using huggingface_hub

Browse files
checkpoint-128/config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "google/vit-base-patch16-224-in21k",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
@@ -9,17 +9,17 @@
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 768,
11
  "id2label": {
12
- "0": "bad_quality",
13
- "1": "good_quality",
14
- "2": "empty_background"
15
  },
16
  "image_size": 224,
17
  "initializer_range": 0.02,
18
  "intermediate_size": 3072,
19
  "label2id": {
20
- "bad_quality": 0,
21
- "empty_background": 2,
22
- "good_quality": 1
23
  },
24
  "layer_norm_eps": 1e-12,
25
  "model_type": "vit",
@@ -30,5 +30,5 @@
30
  "problem_type": "single_label_classification",
31
  "qkv_bias": true,
32
  "torch_dtype": "float32",
33
- "transformers_version": "4.33.3"
34
  }
 
1
  {
2
+ "_name_or_path": "dima806/lemon_quality_image_detection",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
 
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 768,
11
  "id2label": {
12
+ "0": "good_quality",
13
+ "1": "empty_background",
14
+ "2": "bad_quality"
15
  },
16
  "image_size": 224,
17
  "initializer_range": 0.02,
18
  "intermediate_size": 3072,
19
  "label2id": {
20
+ "bad_quality": 2,
21
+ "empty_background": 1,
22
+ "good_quality": 0
23
  },
24
  "layer_norm_eps": 1e-12,
25
  "model_type": "vit",
 
30
  "problem_type": "single_label_classification",
31
  "qkv_bias": true,
32
  "torch_dtype": "float32",
33
+ "transformers_version": "4.39.2"
34
  }
checkpoint-128/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:40cea90ccbf81613a566ed415498d8710e4fef48da79e4b888be8c46c3b28fef
3
+ size 343227052
checkpoint-128/optimizer.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f32d84afd8e56a2cde9e6d2ef722c27411766770b925eb6420f9465e44ec159b
3
  size 686574597
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7bb7ea6794777afd9786c8544c228974bb984ade71f338203e82b2050fea4682
3
  size 686574597
checkpoint-128/preprocessor_config.json CHANGED
@@ -1,4 +1,18 @@
1
  {
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  "do_normalize": true,
3
  "do_rescale": true,
4
  "do_resize": true,
 
1
  {
2
+ "_valid_processor_keys": [
3
+ "images",
4
+ "do_resize",
5
+ "size",
6
+ "resample",
7
+ "do_rescale",
8
+ "rescale_factor",
9
+ "do_normalize",
10
+ "image_mean",
11
+ "image_std",
12
+ "return_tensors",
13
+ "data_format",
14
+ "input_data_format"
15
+ ],
16
  "do_normalize": true,
17
  "do_rescale": true,
18
  "do_resize": true,
checkpoint-128/rng_state.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c33047c27d8e9143c9dcf947d37b78ef88270f94db0df0e2fbdd5921c35f2d93
3
  size 14575
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9120c7be6dc7a83a29fbe261cc51f4ec3c951303d1c0e44912c1dfec99016f42
3
  size 14575
checkpoint-128/scheduler.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5d5ed4e4e1b7febcb3a328700315229b1a315093c0872ab2c59a9e3fb7370a10
3
  size 627
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0635e963639eec33518717a9f8e4bab758f1b7e0b7ee1a261608b542bfd2a157
3
  size 627
checkpoint-128/trainer_state.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "best_metric": 0.3760019540786743,
3
  "best_model_checkpoint": "lemon_quality_image_detection/checkpoint-128",
4
  "epoch": 2.0,
5
  "eval_steps": 500,
@@ -10,28 +10,30 @@
10
  "log_history": [
11
  {
12
  "epoch": 1.0,
13
- "eval_accuracy": 0.9209486166007905,
14
- "eval_loss": 0.6092899441719055,
15
- "eval_runtime": 5.2948,
16
- "eval_samples_per_second": 95.566,
17
- "eval_steps_per_second": 12.087,
18
  "step": 64
19
  },
20
  {
21
  "epoch": 2.0,
22
- "eval_accuracy": 0.9723320158102767,
23
- "eval_loss": 0.3760019540786743,
24
- "eval_runtime": 4.9359,
25
- "eval_samples_per_second": 102.515,
26
- "eval_steps_per_second": 12.966,
27
  "step": 128
28
  }
29
  ],
30
  "logging_steps": 500,
31
  "max_steps": 128,
 
32
  "num_train_epochs": 2,
33
  "save_steps": 500,
34
  "total_flos": 3.133804147057705e+17,
 
35
  "trial_name": null,
36
  "trial_params": null
37
  }
 
1
  {
2
+ "best_metric": 0.05322761833667755,
3
  "best_model_checkpoint": "lemon_quality_image_detection/checkpoint-128",
4
  "epoch": 2.0,
5
  "eval_steps": 500,
 
10
  "log_history": [
11
  {
12
  "epoch": 1.0,
13
+ "eval_accuracy": 1.0,
14
+ "eval_loss": 0.054433587938547134,
15
+ "eval_runtime": 5.0385,
16
+ "eval_samples_per_second": 100.427,
17
+ "eval_steps_per_second": 12.702,
18
  "step": 64
19
  },
20
  {
21
  "epoch": 2.0,
22
+ "eval_accuracy": 1.0,
23
+ "eval_loss": 0.05322761833667755,
24
+ "eval_runtime": 5.1226,
25
+ "eval_samples_per_second": 98.779,
26
+ "eval_steps_per_second": 12.494,
27
  "step": 128
28
  }
29
  ],
30
  "logging_steps": 500,
31
  "max_steps": 128,
32
+ "num_input_tokens_seen": 0,
33
  "num_train_epochs": 2,
34
  "save_steps": 500,
35
  "total_flos": 3.133804147057705e+17,
36
+ "train_batch_size": 32,
37
  "trial_name": null,
38
  "trial_params": null
39
  }
checkpoint-128/training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8acf4c65d2cb2b4179d07c7453b68afb1803bfe7b4697053dc36f65d5a5a6049
3
- size 4027
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b6174adc0fb4e35cf1719b1dc931f145857d37f16e89c2cfa8af2fbcb4b830bc
3
+ size 4411
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "google/vit-base-patch16-224-in21k",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
@@ -30,5 +30,5 @@
30
  "problem_type": "single_label_classification",
31
  "qkv_bias": true,
32
  "torch_dtype": "float32",
33
- "transformers_version": "4.33.3"
34
  }
 
1
  {
2
+ "_name_or_path": "dima806/lemon_quality_image_detection",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
 
30
  "problem_type": "single_label_classification",
31
  "qkv_bias": true,
32
  "torch_dtype": "float32",
33
+ "transformers_version": "4.39.2"
34
  }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:40cea90ccbf81613a566ed415498d8710e4fef48da79e4b888be8c46c3b28fef
3
+ size 343227052
preprocessor_config.json CHANGED
@@ -1,4 +1,18 @@
1
  {
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  "do_normalize": true,
3
  "do_rescale": true,
4
  "do_resize": true,
 
1
  {
2
+ "_valid_processor_keys": [
3
+ "images",
4
+ "do_resize",
5
+ "size",
6
+ "resample",
7
+ "do_rescale",
8
+ "rescale_factor",
9
+ "do_normalize",
10
+ "image_mean",
11
+ "image_std",
12
+ "return_tensors",
13
+ "data_format",
14
+ "input_data_format"
15
+ ],
16
  "do_normalize": true,
17
  "do_rescale": true,
18
  "do_resize": true,
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bca7d24ba12ebd121f2cc1d0b08aa48e6aaa46ae85ffca644c2cf819826d5d20
3
- size 4027
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b6174adc0fb4e35cf1719b1dc931f145857d37f16e89c2cfa8af2fbcb4b830bc
3
+ size 4411