hjay commited on
Commit
f4e6df1
·
verified ·
1 Parent(s): a0d2079

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/mobilenet_v2_1.0_224
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
+ - microsoft/cats_vs_dogs
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,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/mobilenet_v2_1.0_224",
3
+ "_num_labels": 2,
4
+ "architectures": [
5
+ "MobileNetV2ForImageClassification"
6
+ ],
7
+ "classifier_dropout_prob": 0.2,
8
+ "depth_divisible_by": 8,
9
+ "depth_multiplier": 1.0,
10
+ "expand_ratio": 6,
11
+ "finegrained_output": true,
12
+ "first_layer_is_expansion": true,
13
+ "hidden_act": "relu6",
14
+ "id2label": {
15
+ "0": "cat",
16
+ "1": "dog"
17
+ },
18
+ "image_size": 224,
19
+ "initializer_range": 0.02,
20
+ "label2id": {
21
+ "cat": 0,
22
+ "dog": 1
23
+ },
24
+ "layer_norm_eps": 0.001,
25
+ "min_depth": 8,
26
+ "model_type": "mobilenet_v2",
27
+ "num_channels": 3,
28
+ "output_stride": 32,
29
+ "problem_type": "single_label_classification",
30
+ "semantic_loss_ignore_index": 255,
31
+ "tf_padding": true,
32
+ "torch_dtype": "float32",
33
+ "transformers_version": "4.46.2"
34
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e6be5f27943c8db1eaea86f02fcc768043ed5a768c1de2ee07ec8aade6c18d89
3
+ size 9080216
preprocessor_config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "crop_size": {
3
+ "height": 224,
4
+ "width": 224
5
+ },
6
+ "do_center_crop": true,
7
+ "do_normalize": true,
8
+ "do_rescale": true,
9
+ "do_resize": true,
10
+ "image_mean": [
11
+ 0.5,
12
+ 0.5,
13
+ 0.5
14
+ ],
15
+ "image_processor_type": "MobileNetV2ImageProcessor",
16
+ "image_std": [
17
+ 0.5,
18
+ 0.5,
19
+ 0.5
20
+ ],
21
+ "resample": 2,
22
+ "rescale_factor": 0.00392156862745098,
23
+ "size": {
24
+ "shortest_edge": 256
25
+ }
26
+ }
runs/Nov22_11-11-13_r-hjay-autotrain-advanced-yx93g6ph-b1ec0-6e1jf/events.out.tfevents.1732273875.r-hjay-autotrain-advanced-yx93g6ph-b1ec0-6e1jf.136.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:20b70d053cd5b1db982fa7c6768c35a5c65c47e7710b8f86cf05da9b6b37927a
3
- size 76257
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d2c9b5f59ec209205d9c9b0336f55bca61cfaca9e2fb0b8a2487f618ac26a601
3
+ size 79565
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9f8367e0766bb269c2d3814b0920ff1aadd3ca9bc4bc1ebce4c5cdd571cef67f
3
+ size 5304
training_params.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "microsoft/cats_vs_dogs",
3
+ "model": "google/mobilenet_v2_1.0_224",
4
+ "username": "hjay",
5
+ "lr": 5e-05,
6
+ "epochs": 3,
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": "autotrain-z7ygf-g8xy8",
19
+ "auto_find_batch_size": false,
20
+ "mixed_precision": "fp16",
21
+ "save_total_limit": 1,
22
+ "push_to_hub": true,
23
+ "eval_strategy": "epoch",
24
+ "image_column": "image",
25
+ "target_column": "labels",
26
+ "log": "tensorboard",
27
+ "early_stopping_patience": 5,
28
+ "early_stopping_threshold": 0.01
29
+ }