ThatOrJohn commited on
Commit
9c6e05f
·
verified ·
1 Parent(s): 7412385

Upload folder using huggingface_hub

Browse files
README.md ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ tags:
4
+ - autotrain
5
+ - image-classification
6
+ base_model: google/efficientnet-b1
7
+ widget:
8
+ - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/tiger.jpg
9
+ example_title: Tiger
10
+ - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/teapot.jpg
11
+ example_title: Teapot
12
+ - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/palace.jpg
13
+ example_title: Palace
14
+ datasets:
15
+ - ThatOrJohn/pineapple
16
+ ---
17
+
18
+ # Model Trained Using AutoTrain
19
+
20
+ - Problem type: Image Classification
21
+
22
+ ## Validation Metrics
23
+ No validation metrics available
config.json ADDED
@@ -0,0 +1,91 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/efficientnet-b1",
3
+ "_num_labels": 2,
4
+ "architectures": [
5
+ "EfficientNetForImageClassification"
6
+ ],
7
+ "batch_norm_eps": 0.001,
8
+ "batch_norm_momentum": 0.99,
9
+ "depth_coefficient": 1.1,
10
+ "depth_divisor": 8,
11
+ "depthwise_padding": [
12
+ 16
13
+ ],
14
+ "drop_connect_rate": 0.2,
15
+ "dropout_rate": 0.2,
16
+ "expand_ratios": [
17
+ 1,
18
+ 6,
19
+ 6,
20
+ 6,
21
+ 6,
22
+ 6,
23
+ 6
24
+ ],
25
+ "hidden_act": "swish",
26
+ "hidden_dim": 1280,
27
+ "id2label": {
28
+ "0": "pineapple-chunk",
29
+ "1": "pineapple-ring"
30
+ },
31
+ "image_size": 240,
32
+ "in_channels": [
33
+ 32,
34
+ 16,
35
+ 24,
36
+ 40,
37
+ 80,
38
+ 112,
39
+ 192
40
+ ],
41
+ "initializer_range": 0.02,
42
+ "kernel_sizes": [
43
+ 3,
44
+ 3,
45
+ 5,
46
+ 3,
47
+ 5,
48
+ 5,
49
+ 3
50
+ ],
51
+ "label2id": {
52
+ "pineapple-chunk": 0,
53
+ "pineapple-ring": 1
54
+ },
55
+ "model_type": "efficientnet",
56
+ "num_block_repeats": [
57
+ 1,
58
+ 2,
59
+ 2,
60
+ 3,
61
+ 3,
62
+ 4,
63
+ 1
64
+ ],
65
+ "num_channels": 3,
66
+ "num_hidden_layers": 64,
67
+ "out_channels": [
68
+ 16,
69
+ 24,
70
+ 40,
71
+ 80,
72
+ 112,
73
+ 192,
74
+ 320
75
+ ],
76
+ "pooling_type": "mean",
77
+ "problem_type": "single_label_classification",
78
+ "squeeze_expansion_ratio": 0.25,
79
+ "strides": [
80
+ 1,
81
+ 2,
82
+ 2,
83
+ 2,
84
+ 1,
85
+ 2,
86
+ 1
87
+ ],
88
+ "torch_dtype": "float32",
89
+ "transformers_version": "4.41.2",
90
+ "width_coefficient": 1.0
91
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:efe30da6b648e7a0e04fac9697364c5671f3019859a22fa7f615789177056846
3
+ size 26377248
preprocessor_config.json ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_valid_processor_keys": [
3
+ "images",
4
+ "do_resize",
5
+ "size",
6
+ "resample",
7
+ "do_center_crop",
8
+ "crop_size",
9
+ "do_rescale",
10
+ "rescale_factor",
11
+ "rescale_offset",
12
+ "do_normalize",
13
+ "image_mean",
14
+ "image_std",
15
+ "include_top",
16
+ "return_tensors",
17
+ "data_format",
18
+ "input_data_format"
19
+ ],
20
+ "crop_size": {
21
+ "height": 289,
22
+ "width": 289
23
+ },
24
+ "do_center_crop": false,
25
+ "do_normalize": true,
26
+ "do_rescale": true,
27
+ "do_resize": true,
28
+ "image_mean": [
29
+ 0.485,
30
+ 0.456,
31
+ 0.406
32
+ ],
33
+ "image_processor_type": "EfficientNetImageProcessor",
34
+ "image_std": [
35
+ 0.47853944,
36
+ 0.4732864,
37
+ 0.47434163
38
+ ],
39
+ "include_top": true,
40
+ "resample": 0,
41
+ "rescale_factor": 0.00392156862745098,
42
+ "rescale_offset": false,
43
+ "size": {
44
+ "height": 240,
45
+ "width": 240
46
+ }
47
+ }
runs/Jun12_18-50-12_Johns-Mac-mini.attlocal.net/events.out.tfevents.1718236213.Johns-Mac-mini.attlocal.net.82231.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a8b59dd37566d44875609c9e6fdd3edbdc32b9ad93414045f1e26ed731afc41e
3
- size 5458
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:44ddc4472dba37cfe645650adf0ce6b5a1c53dea100d1791c469e7d209d9d3b3
3
+ size 11602
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:427f9f4c5d7967050e88ec545d61914b76158ccde7b04f5b33cc54cacd96da6b
3
+ size 5112
training_params.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "ThatOrJohn/pineapple",
3
+ "model": "google/efficientnet-b1",
4
+ "username": "ThatOrJohn",
5
+ "lr": 5e-05,
6
+ "epochs": 4,
7
+ "batch_size": 8,
8
+ "warmup_ratio": 0.1,
9
+ "gradient_accumulation": 1,
10
+ "optimizer": "adamw_torch",
11
+ "scheduler": "linear",
12
+ "weight_decay": 0.0,
13
+ "max_grad_norm": 1.0,
14
+ "seed": 42,
15
+ "train_split": "train",
16
+ "valid_split": null,
17
+ "logging_steps": -1,
18
+ "project_name": "efficientnet-b1-pineapple",
19
+ "auto_find_batch_size": false,
20
+ "mixed_precision": "none",
21
+ "save_total_limit": 1,
22
+ "push_to_hub": true,
23
+ "evaluation_strategy": "epoch",
24
+ "image_column": "image",
25
+ "target_column": "label",
26
+ "log": "tensorboard",
27
+ "early_stopping_patience": 5,
28
+ "early_stopping_threshold": 0.01
29
+ }