Iust1n2/WikiArtWithResNet18

README.md

@@ -0,0 +1,62 @@
+---
+license: apache-2.0
+base_model: microsoft/resnet-18
+tags:
+- generated_from_trainer
+metrics:
+- accuracy
+model-index:
+- name: resnet-18-finetuned-wikiart
+  results: []
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+# resnet-18-finetuned-wikiart
+
+This model is a fine-tuned version of [microsoft/resnet-18](https://huggingface.co/microsoft/resnet-18) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 1.6256
+- Accuracy: 0.6247
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 2e-05
+- train_batch_size: 32
+- eval_batch_size: 32
+- seed: 42
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: linear
+- num_epochs: 3
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Accuracy |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 2.0382        | 1.0   | 2037 | 1.8771          | 0.5938   |
+| 1.8027        | 2.0   | 4074 | 1.6860          | 0.6160   |
+| 1.7033        | 3.0   | 6111 | 1.6256          | 0.6247   |
+
+
+### Framework versions
+
+- Transformers 4.35.2
+- Pytorch 2.1.1+cu121
+- Datasets 2.15.0
+- Tokenizers 0.15.0

all_results.json

@@ -0,0 +1,13 @@
+{
+    "epoch": 3.0,
+    "eval_accuracy": 0.6246546749340045,
+    "eval_loss": 1.6256191730499268,
+    "eval_runtime": 294.5998,
+    "eval_samples_per_second": 55.292,
+    "eval_steps_per_second": 1.731,
+    "total_flos": 1.9847588302331597e+18,
+    "train_loss": 2.015392058142623,
+    "train_runtime": 3974.3228,
+    "train_samples_per_second": 49.182,
+    "train_steps_per_second": 1.538
+}

config.json

@@ -0,0 +1,303 @@
+{
+  "_name_or_path": "microsoft/resnet-18",
+  "architectures": [
+    "ResNetForImageClassification"
+  ],
+  "depths": [
+    2,
+    2,
+    2,
+    2
+  ],
+  "downsample_in_bottleneck": false,
+  "downsample_in_first_stage": false,
+  "embedding_size": 64,
+  "hidden_act": "relu",
+  "hidden_sizes": [
+    64,
+    128,
+    256,
+    512
+  ],
+  "id2label": {
+    "0": "Unknown Artist",
+    "1": "boris-kustodiev",
+    "2": "camille-pissarro",
+    "3": "childe-hassam",
+    "4": "claude-monet",
+    "5": "edgar-degas",
+    "6": "eugene-boudin",
+    "7": "gustave-dore",
+    "8": "ilya-repin",
+    "9": "ivan-aivazovsky",
+    "10": "ivan-shishkin",
+    "11": "john-singer-sargent",
+    "12": "marc-chagall",
+    "13": "martiros-saryan",
+    "14": "nicholas-roerich",
+    "15": "pablo-picasso",
+    "16": "paul-cezanne",
+    "17": "pierre-auguste-renoir",
+    "18": "pyotr-konchalovsky",
+    "19": "raphael-kirchner",
+    "20": "rembrandt",
+    "21": "salvador-dali",
+    "22": "vincent-van-gogh",
+    "23": "hieronymus-bosch",
+    "24": "leonardo-da-vinci",
+    "25": "albrecht-durer",
+    "26": "edouard-cortes",
+    "27": "sam-francis",
+    "28": "juan-gris",
+    "29": "lucas-cranach-the-elder",
+    "30": "paul-gauguin",
+    "31": "konstantin-makovsky",
+    "32": "egon-schiele",
+    "33": "thomas-eakins",
+    "34": "gustave-moreau",
+    "35": "francisco-goya",
+    "36": "edvard-munch",
+    "37": "henri-matisse",
+    "38": "fra-angelico",
+    "39": "maxime-maufra",
+    "40": "jan-matejko",
+    "41": "mstislav-dobuzhinsky",
+    "42": "alfred-sisley",
+    "43": "mary-cassatt",
+    "44": "gustave-loiseau",
+    "45": "fernando-botero",
+    "46": "zinaida-serebriakova",
+    "47": "georges-seurat",
+    "48": "isaac-levitan",
+    "49": "joaqu\u00e3\u00adn-sorolla",
+    "50": "jacek-malczewski",
+    "51": "berthe-morisot",
+    "52": "andy-warhol",
+    "53": "arkhip-kuindzhi",
+    "54": "niko-pirosmani",
+    "55": "james-tissot",
+    "56": "vasily-polenov",
+    "57": "valentin-serov",
+    "58": "pietro-perugino",
+    "59": "pierre-bonnard",
+    "60": "ferdinand-hodler",
+    "61": "bartolome-esteban-murillo",
+    "62": "giovanni-boldini",
+    "63": "henri-martin",
+    "64": "gustav-klimt",
+    "65": "vasily-perov",
+    "66": "odilon-redon",
+    "67": "tintoretto",
+    "68": "gene-davis",
+    "69": "raphael",
+    "70": "john-henry-twachtman",
+    "71": "henri-de-toulouse-lautrec",
+    "72": "antoine-blanchard",
+    "73": "david-burliuk",
+    "74": "camille-corot",
+    "75": "konstantin-korovin",
+    "76": "ivan-bilibin",
+    "77": "titian",
+    "78": "maurice-prendergast",
+    "79": "edouard-manet",
+    "80": "peter-paul-rubens",
+    "81": "aubrey-beardsley",
+    "82": "paolo-veronese",
+    "83": "joshua-reynolds",
+    "84": "kuzma-petrov-vodkin",
+    "85": "gustave-caillebotte",
+    "86": "lucian-freud",
+    "87": "michelangelo",
+    "88": "dante-gabriel-rossetti",
+    "89": "felix-vallotton",
+    "90": "nikolay-bogdanov-belsky",
+    "91": "georges-braque",
+    "92": "vasily-surikov",
+    "93": "fernand-leger",
+    "94": "konstantin-somov",
+    "95": "katsushika-hokusai",
+    "96": "sir-lawrence-alma-tadema",
+    "97": "vasily-vereshchagin",
+    "98": "ernst-ludwig-kirchner",
+    "99": "mikhail-vrubel",
+    "100": "orest-kiprensky",
+    "101": "william-merritt-chase",
+    "102": "aleksey-savrasov",
+    "103": "hans-memling",
+    "104": "amedeo-modigliani",
+    "105": "ivan-kramskoy",
+    "106": "utagawa-kuniyoshi",
+    "107": "gustave-courbet",
+    "108": "william-turner",
+    "109": "theo-van-rysselberghe",
+    "110": "joseph-wright",
+    "111": "edward-burne-jones",
+    "112": "koloman-moser",
+    "113": "viktor-vasnetsov",
+    "114": "anthony-van-dyck",
+    "115": "raoul-dufy",
+    "116": "frans-hals",
+    "117": "hans-holbein-the-younger",
+    "118": "ilya-mashkov",
+    "119": "henri-fantin-latour",
+    "120": "m.c.-escher",
+    "121": "el-greco",
+    "122": "mikalojus-ciurlionis",
+    "123": "james-mcneill-whistler",
+    "124": "karl-bryullov",
+    "125": "jacob-jordaens",
+    "126": "thomas-gainsborough",
+    "127": "eugene-delacroix",
+    "128": "canaletto"
+  },
+  "label2id": {
+    "Unknown Artist": 0,
+    "albrecht-durer": 25,
+    "aleksey-savrasov": 102,
+    "alfred-sisley": 42,
+    "amedeo-modigliani": 104,
+    "andy-warhol": 52,
+    "anthony-van-dyck": 114,
+    "antoine-blanchard": 72,
+    "arkhip-kuindzhi": 53,
+    "aubrey-beardsley": 81,
+    "bartolome-esteban-murillo": 61,
+    "berthe-morisot": 51,
+    "boris-kustodiev": 1,
+    "camille-corot": 74,
+    "camille-pissarro": 2,
+    "canaletto": 128,
+    "childe-hassam": 3,
+    "claude-monet": 4,
+    "dante-gabriel-rossetti": 88,
+    "david-burliuk": 73,
+    "edgar-degas": 5,
+    "edouard-cortes": 26,
+    "edouard-manet": 79,
+    "edvard-munch": 36,
+    "edward-burne-jones": 111,
+    "egon-schiele": 32,
+    "el-greco": 121,
+    "ernst-ludwig-kirchner": 98,
+    "eugene-boudin": 6,
+    "eugene-delacroix": 127,
+    "felix-vallotton": 89,
+    "ferdinand-hodler": 60,
+    "fernand-leger": 93,
+    "fernando-botero": 45,
+    "fra-angelico": 38,
+    "francisco-goya": 35,
+    "frans-hals": 116,
+    "gene-davis": 68,
+    "georges-braque": 91,
+    "georges-seurat": 47,
+    "giovanni-boldini": 62,
+    "gustav-klimt": 64,
+    "gustave-caillebotte": 85,
+    "gustave-courbet": 107,
+    "gustave-dore": 7,
+    "gustave-loiseau": 44,
+    "gustave-moreau": 34,
+    "hans-holbein-the-younger": 117,
+    "hans-memling": 103,
+    "henri-de-toulouse-lautrec": 71,
+    "henri-fantin-latour": 119,
+    "henri-martin": 63,
+    "henri-matisse": 37,
+    "hieronymus-bosch": 23,
+    "ilya-mashkov": 118,
+    "ilya-repin": 8,
+    "isaac-levitan": 48,
+    "ivan-aivazovsky": 9,
+    "ivan-bilibin": 76,
+    "ivan-kramskoy": 105,
+    "ivan-shishkin": 10,
+    "jacek-malczewski": 50,
+    "jacob-jordaens": 125,
+    "james-mcneill-whistler": 123,
+    "james-tissot": 55,
+    "jan-matejko": 40,
+    "joaqu\u00e3\u00adn-sorolla": 49,
+    "john-henry-twachtman": 70,
+    "john-singer-sargent": 11,
+    "joseph-wright": 110,
+    "joshua-reynolds": 83,
+    "juan-gris": 28,
+    "karl-bryullov": 124,
+    "katsushika-hokusai": 95,
+    "koloman-moser": 112,
+    "konstantin-korovin": 75,
+    "konstantin-makovsky": 31,
+    "konstantin-somov": 94,
+    "kuzma-petrov-vodkin": 84,
+    "leonardo-da-vinci": 24,
+    "lucas-cranach-the-elder": 29,
+    "lucian-freud": 86,
+    "m.c.-escher": 120,
+    "marc-chagall": 12,
+    "martiros-saryan": 13,
+    "mary-cassatt": 43,
+    "maurice-prendergast": 78,
+    "maxime-maufra": 39,
+    "michelangelo": 87,
+    "mikalojus-ciurlionis": 122,
+    "mikhail-vrubel": 99,
+    "mstislav-dobuzhinsky": 41,
+    "nicholas-roerich": 14,
+    "niko-pirosmani": 54,
+    "nikolay-bogdanov-belsky": 90,
+    "odilon-redon": 66,
+    "orest-kiprensky": 100,
+    "pablo-picasso": 15,
+    "paolo-veronese": 82,
+    "paul-cezanne": 16,
+    "paul-gauguin": 30,
+    "peter-paul-rubens": 80,
+    "pierre-auguste-renoir": 17,
+    "pierre-bonnard": 59,
+    "pietro-perugino": 58,
+    "pyotr-konchalovsky": 18,
+    "raoul-dufy": 115,
+    "raphael": 69,
+    "raphael-kirchner": 19,
+    "rembrandt": 20,
+    "salvador-dali": 21,
+    "sam-francis": 27,
+    "sir-lawrence-alma-tadema": 96,
+    "theo-van-rysselberghe": 109,
+    "thomas-eakins": 33,
+    "thomas-gainsborough": 126,
+    "tintoretto": 67,
+    "titian": 77,
+    "utagawa-kuniyoshi": 106,
+    "valentin-serov": 57,
+    "vasily-perov": 65,
+    "vasily-polenov": 56,
+    "vasily-surikov": 92,
+    "vasily-vereshchagin": 97,
+    "viktor-vasnetsov": 113,
+    "vincent-van-gogh": 22,
+    "william-merritt-chase": 101,
+    "william-turner": 108,
+    "zinaida-serebriakova": 46
+  },
+  "layer_type": "basic",
+  "model_type": "resnet",
+  "num_channels": 3,
+  "out_features": [
+    "stage4"
+  ],
+  "out_indices": [
+    4
+  ],
+  "problem_type": "single_label_classification",
+  "stage_names": [
+    "stem",
+    "stage1",
+    "stage2",
+    "stage3",
+    "stage4"
+  ],
+  "torch_dtype": "float32",
+  "transformers_version": "4.35.2"
+}

eval_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 3.0,
+    "eval_accuracy": 0.6246546749340045,
+    "eval_loss": 1.6256191730499268,
+    "eval_runtime": 294.5998,
+    "eval_samples_per_second": 55.292,
+    "eval_steps_per_second": 1.731
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4fa7606c3761ec6ece0dcc0cc6c82347dfeeedd9a315e5dff039aae0528f7e03
+size 45024964

preprocessor_config.json

@@ -0,0 +1,22 @@
+{
+  "crop_pct": 0.875,
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "image_mean": [
+    0.485,
+    0.456,
+    0.406
+  ],
+  "image_processor_type": "ConvNextImageProcessor",
+  "image_std": [
+    0.229,
+    0.224,
+    0.225
+  ],
+  "resample": 3,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "shortest_edge": 224
+  }
+}

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 3.0,
+    "total_flos": 1.9847588302331597e+18,
+    "train_loss": 2.015392058142623,
+    "train_runtime": 3974.3228,
+    "train_samples_per_second": 49.182,
+    "train_steps_per_second": 1.538
+}

trainer_state.json

@@ -0,0 +1,127 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 3.0,
+  "eval_steps": 500,
+  "global_step": 6111,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.25,
+      "learning_rate": 1.836360661102929e-05,
+      "loss": 3.3023,
+      "step": 500
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 1.6727213222058585e-05,
+      "loss": 2.3284,
+      "step": 1000
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 1.5090819833087874e-05,
+      "loss": 2.1442,
+      "step": 1500
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 1.3454426444117168e-05,
+      "loss": 2.0382,
+      "step": 2000
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.5937749401436552,
+      "eval_loss": 1.8770618438720703,
+      "eval_runtime": 302.3493,
+      "eval_samples_per_second": 53.875,
+      "eval_steps_per_second": 1.687,
+      "step": 2037
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 1.1818033055146459e-05,
+      "loss": 1.9449,
+      "step": 2500
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 1.0181639666175749e-05,
+      "loss": 1.8855,
+      "step": 3000
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 8.54524627720504e-06,
+      "loss": 1.839,
+      "step": 3500
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 6.908852888234332e-06,
+      "loss": 1.8027,
+      "step": 4000
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.6159985266130518,
+      "eval_loss": 1.6860090494155884,
+      "eval_runtime": 293.1285,
+      "eval_samples_per_second": 55.569,
+      "eval_steps_per_second": 1.74,
+      "step": 4074
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 5.272459499263623e-06,
+      "loss": 1.7814,
+      "step": 4500
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 3.6360661102929145e-06,
+      "loss": 1.7355,
+      "step": 5000
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 1.999672721322206e-06,
+      "loss": 1.7351,
+      "step": 5500
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 3.6327933235149736e-07,
+      "loss": 1.7033,
+      "step": 6000
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.6246546749340045,
+      "eval_loss": 1.6256191730499268,
+      "eval_runtime": 298.8328,
+      "eval_samples_per_second": 54.509,
+      "eval_steps_per_second": 1.707,
+      "step": 6111
+    },
+    {
+      "epoch": 3.0,
+      "step": 6111,
+      "total_flos": 1.9847588302331597e+18,
+      "train_loss": 2.015392058142623,
+      "train_runtime": 3974.3228,
+      "train_samples_per_second": 49.182,
+      "train_steps_per_second": 1.538
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 6111,
+  "num_train_epochs": 3,
+  "save_steps": 500,
+  "total_flos": 1.9847588302331597e+18,
+  "trial_name": null,
+  "trial_params": null
+}

training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:41a73490a7795370db893b7038d3bef8b0a55aca618ee3b17f8a49f683edec00
+size 4600