Upload folder using huggingface_hub

README.md

@@ -0,0 +1,76 @@
+---
+base_model: microsoft/resnet-18
+library_name: transformers
+tags:
+- fusion-bench
+- merge
+---
+# Deep Model Fusion
+
+Fine-tuned ResNet model on dataset gtsrb.
+
+## Models Merged
+
+This is a merged model created using [fusion-bench](https://github.com/tanganke/fusion_bench).
+
+The following models were included in the merge:
+
+
+
+
+## Configuration
+
+The following YAML configuration was used to produce this model:
+
+### Algorithm Configuration
+
+```yaml
+_recursive_: false
+_target_: fusion_bench.method.classification.image_classification_finetune.ImageClassificationFineTuning
+_usage_: null
+_version_: 0.2.25.dev0
+dataloader_kwargs:
+  batch_size: 128
+  num_workers: 8
+  pin_memory: true
+label_smoothing: 0
+lr_scheduler: null
+max_epochs: -1
+max_steps: 4000
+optimizer:
+  _target_: torch.optim.SGD
+  lr: 0.001
+  momentum: 0.9
+  weight_decay: 0.0001
+save_interval: 1000
+save_on_train_epoch_end: false
+save_top_k: -1
+training_data_ratio: 0.5
+```
+
+### Model Pool Configuration
+
+```yaml
+_recursive_: false
+_target_: fusion_bench.modelpool.resnet_for_image_classification.ResNetForImageClassificationPool
+_usage_: null
+_version_: 0.2.25.dev0
+models:
+  _pretrained_:
+    config_path: microsoft/resnet-18
+    dataset_name: gtsrb
+    pretrained: true
+test_datasets: null
+train_datasets:
+  gtsrb:
+    _target_: datasets.load_dataset
+    path: tanganke/gtsrb
+    split: train
+type: transformers
+val_datasets:
+  gtsrb:
+    _target_: datasets.load_dataset
+    path: tanganke/gtsrb
+    split: test
+```
+

config.json

@@ -0,0 +1,129 @@
+{
+  "architectures": [
+    "ResNetForImageClassification"
+  ],
+  "depths": [
+    2,
+    2,
+    2,
+    2
+  ],
+  "downsample_in_bottleneck": false,
+  "downsample_in_first_stage": false,
+  "dtype": "float32",
+  "embedding_size": 64,
+  "hidden_act": "relu",
+  "hidden_sizes": [
+    64,
+    128,
+    256,
+    512
+  ],
+  "id2label": {
+    "0": "red and white circle 20 kph speed limit",
+    "1": "red and white circle 30 kph speed limit",
+    "2": "red and white circle 50 kph speed limit",
+    "3": "red and white circle 60 kph speed limit",
+    "4": "red and white circle 70 kph speed limit",
+    "5": "red and white circle 80 kph speed limit",
+    "6": "end / de-restriction of 80 kph speed limit",
+    "7": "red and white circle 100 kph speed limit",
+    "8": "red and white circle 120 kph speed limit",
+    "9": "red and white circle red car and black car no passing",
+    "10": "red and white circle red truck and black car no passing",
+    "11": "red and white triangle road intersection warning",
+    "12": "white and yellow diamond priority road",
+    "13": "red and white upside down triangle yield right-of-way",
+    "14": "stop",
+    "15": "empty red and white circle",
+    "16": "red and white circle no truck entry",
+    "17": "red circle with white horizonal stripe no entry",
+    "18": "red and white triangle with exclamation mark warning",
+    "19": "red and white triangle with black left curve approaching warning",
+    "20": "red and white triangle with black right curve approaching warning",
+    "21": "red and white triangle with black double curve approaching warning",
+    "22": "red and white triangle rough / bumpy road warning",
+    "23": "red and white triangle car skidding / slipping warning",
+    "24": "red and white triangle with merging / narrow lanes warning",
+    "25": "red and white triangle with person digging / construction / road work warning",
+    "26": "red and white triangle with traffic light approaching warning",
+    "27": "red and white triangle with person walking warning",
+    "28": "red and white triangle with child and person walking warning",
+    "29": "red and white triangle with bicyle warning",
+    "30": "red and white triangle with snowflake / ice warning",
+    "31": "red and white triangle with deer warning",
+    "32": "white circle with gray strike bar no speed limit",
+    "33": "blue circle with white right turn arrow mandatory",
+    "34": "blue circle with white left turn arrow mandatory",
+    "35": "blue circle with white forward arrow mandatory",
+    "36": "blue circle with white forward or right turn arrow mandatory",
+    "37": "blue circle with white forward or left turn arrow mandatory",
+    "38": "blue circle with white keep right arrow mandatory",
+    "39": "blue circle with white keep left arrow mandatory",
+    "40": "blue circle with white arrows indicating a traffic circle",
+    "41": "white circle with gray strike bar indicating no passing for cars has ended",
+    "42": "white circle with gray strike bar indicating no passing for trucks has ended"
+  },
+  "label2id": {
+    "blue circle with white arrows indicating a traffic circle": 40,
+    "blue circle with white forward arrow mandatory": 35,
+    "blue circle with white forward or left turn arrow mandatory": 37,
+    "blue circle with white forward or right turn arrow mandatory": 36,
+    "blue circle with white keep left arrow mandatory": 39,
+    "blue circle with white keep right arrow mandatory": 38,
+    "blue circle with white left turn arrow mandatory": 34,
+    "blue circle with white right turn arrow mandatory": 33,
+    "empty red and white circle": 15,
+    "end / de-restriction of 80 kph speed limit": 6,
+    "red and white circle 100 kph speed limit": 7,
+    "red and white circle 120 kph speed limit": 8,
+    "red and white circle 20 kph speed limit": 0,
+    "red and white circle 30 kph speed limit": 1,
+    "red and white circle 50 kph speed limit": 2,
+    "red and white circle 60 kph speed limit": 3,
+    "red and white circle 70 kph speed limit": 4,
+    "red and white circle 80 kph speed limit": 5,
+    "red and white circle no truck entry": 16,
+    "red and white circle red car and black car no passing": 9,
+    "red and white circle red truck and black car no passing": 10,
+    "red and white triangle car skidding / slipping warning": 23,
+    "red and white triangle road intersection warning": 11,
+    "red and white triangle rough / bumpy road warning": 22,
+    "red and white triangle with bicyle warning": 29,
+    "red and white triangle with black double curve approaching warning": 21,
+    "red and white triangle with black left curve approaching warning": 19,
+    "red and white triangle with black right curve approaching warning": 20,
+    "red and white triangle with child and person walking warning": 28,
+    "red and white triangle with deer warning": 31,
+    "red and white triangle with exclamation mark warning": 18,
+    "red and white triangle with merging / narrow lanes warning": 24,
+    "red and white triangle with person digging / construction / road work warning": 25,
+    "red and white triangle with person walking warning": 27,
+    "red and white triangle with snowflake / ice warning": 30,
+    "red and white triangle with traffic light approaching warning": 26,
+    "red and white upside down triangle yield right-of-way": 13,
+    "red circle with white horizonal stripe no entry": 17,
+    "stop": 14,
+    "white and yellow diamond priority road": 12,
+    "white circle with gray strike bar indicating no passing for cars has ended": 41,
+    "white circle with gray strike bar indicating no passing for trucks has ended": 42,
+    "white circle with gray strike bar no speed limit": 32
+  },
+  "layer_type": "basic",
+  "model_type": "resnet",
+  "num_channels": 3,
+  "out_features": [
+    "stage4"
+  ],
+  "out_indices": [
+    4
+  ],
+  "stage_names": [
+    "stem",
+    "stage1",
+    "stage2",
+    "stage3",
+    "stage4"
+  ],
+  "transformers_version": "4.56.1"
+}

model.safetensors

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

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
+  }
+}