ddpm-butterflies-128 / config.json
shamaseen's picture
Training in progress, epoch 0
10490d5
{
"_name_or_path": "google/efficientnet-b3",
"architectures": [
"EfficientNetForImageClassification"
],
"batch_norm_eps": 0.001,
"batch_norm_momentum": 0.99,
"depth_coefficient": 1.4,
"depth_divisor": 8,
"depthwise_padding": [
5,
18
],
"drop_connect_rate": 0.2,
"dropout_rate": 0.3,
"expand_ratios": [
1,
6,
6,
6,
6,
6,
6
],
"hidden_act": "swish",
"hidden_dim": 1536,
"id2label": {
"0": "seat_belt",
"1": "no_seat_belt"
},
"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": {
"no_seat_belt": 1,
"seat_belt": 0
},
"model_type": "efficientnet",
"num_block_repeats": [
1,
2,
2,
3,
3,
4,
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": [
1,
2,
2,
2,
1,
2,
1
],
"torch_dtype": "float32",
"transformers_version": "4.27.3",
"width_coefficient": 1.2
}