Training in progress, epoch 1

config.json

@@ -1,32 +1,29 @@
 {
-  "_name_or_path": "nvidia/mit-b0",
+  "_name_or_path": "google/efficientnet-b3",
   "architectures": [
-    "SegformerForImageClassification"
+    "EfficientNetForImageClassification"
   ],
-  "attention_probs_dropout_prob": 0.0,
-  "classifier_dropout_prob": 0.1,
-  "decoder_hidden_size": 256,
-  "depths": [
-    2,
-    2,
-    2,
-    2
+  "batch_norm_eps": 0.001,
+  "batch_norm_momentum": 0.99,
+  "depth_coefficient": 1.4,
+  "depth_divisor": 8,
+  "depthwise_padding": [
+    5,
+    18
   ],
-  "downsampling_rates": [
+  "drop_connect_rate": 0.2,
+  "dropout_rate": 0.3,
+  "expand_ratios": [
     1,
-    4,
-    8,
-    16
-  ],
-  "drop_path_rate": 0.1,
-  "hidden_act": "gelu",
-  "hidden_dropout_prob": 0.0,
-  "hidden_sizes": [
-    32,
-    64,
-    160,
-    256
+    6,
+    6,
+    6,
+    6,
+    6,
+    6
   ],
+  "hidden_act": "swish",
+  "hidden_dim": 1536,
   "id2label": {
     "0": "Achaemenid architecture",
     "1": "American Foursquare architecture",
@@ -54,8 +51,26 @@
     "8": "Beaux-Arts architecture",
     "9": "Byzantine architecture"
   },
-  "image_size": 224,
+  "image_size": 300,
+  "in_channels": [
+    32,
+    16,
+    24,
+    40,
+    80,
+    112,
+    192
+  ],
   "initializer_range": 0.02,
+  "kernel_sizes": [
+    3,
+    3,
+    5,
+    3,
+    5,
+    5,
+    3
+  ],
   "label2id": {
     "Achaemenid architecture": "0",
     "American Foursquare architecture": "1",
@@ -83,43 +98,40 @@
     "Russian Revival architecture": "23",
     "Tudor Revival architecture": "24"
   },
-  "layer_norm_eps": 1e-06,
-  "mlp_ratios": [
-    4,
-    4,
-    4,
-    4
-  ],
-  "model_type": "segformer",
-  "num_attention_heads": [
+  "model_type": "efficientnet",
+  "num_block_repeats": [
     1,
     2,
-    5,
-    8
-  ],
-  "num_channels": 3,
-  "num_encoder_blocks": 4,
-  "patch_sizes": [
-    7,
+    2,
     3,
     3,
-    3
-  ],
-  "problem_type": "single_label_classification",
-  "reshape_last_stage": true,
-  "semantic_loss_ignore_index": 255,
-  "sr_ratios": [
-    8,
     4,
-    2,
     1
   ],
+  "num_channels": 3,
+  "num_hidden_layers": 64,
+  "out_channels": [
+    16,
+    24,
+    40,
+    80,
+    112,
+    192,
+    320
+  ],
+  "pooling_type": "mean",
+  "problem_type": "single_label_classification",
+  "squeeze_expansion_ratio": 0.25,
   "strides": [
-    4,
+    1,
     2,
     2,
-    2
+    2,
+    1,
+    2,
+    1
   ],
   "torch_dtype": "float32",
-  "transformers_version": "4.41.1"
+  "transformers_version": "4.41.1",
+  "width_coefficient": 1.2
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d28440ec9dedd95e1609a624df0ba59641d78007c084b295ba6df50f6eb0e12f
-size 13325828
+oid sha256:d1450ccc8564dbc7db51931e16500f947494589a6b3e5b4affec5158d0c0c2af
+size 43362788

preprocessor_config.json

@@ -1,22 +1,28 @@
 {
   "_valid_processor_keys": [
     "images",
-    "segmentation_maps",
     "do_resize",
     "size",
     "resample",
+    "do_center_crop",
+    "crop_size",
     "do_rescale",
     "rescale_factor",
+    "rescale_offset",
     "do_normalize",
     "image_mean",
     "image_std",
-    "do_reduce_labels",
+    "include_top",
     "return_tensors",
     "data_format",
     "input_data_format"
   ],
+  "crop_size": {
+    "height": 289,
+    "width": 289
+  },
+  "do_center_crop": false,
   "do_normalize": true,
-  "do_reduce_labels": false,
   "do_rescale": true,
   "do_resize": true,
   "image_mean": [
@@ -24,16 +30,18 @@
     0.456,
     0.406
   ],
-  "image_processor_type": "SegformerImageProcessor",
+  "image_processor_type": "EfficientNetImageProcessor",
   "image_std": [
-    0.229,
-    0.224,
-    0.225
+    0.47853944,
+    0.4732864,
+    0.47434163
   ],
-  "resample": 2,
+  "include_top": true,
+  "resample": 0,
   "rescale_factor": 0.00392156862745098,
+  "rescale_offset": false,
   "size": {
-    "height": 512,
-    "width": 512
+    "height": 300,
+    "width": 300
   }
 }

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:98786f1cd4c0b028594b6612e7d65db2954ee3d008d0a3f7c66820735d7fdba0
+oid sha256:b2cd1c81545de728925a469fc7623f2303bfb084ecfc024cebf24cdf7c754428
 size 5112