test / config.json
james-burton's picture
End of training
3752cdf verified
{
"_name_or_path": "../models/james-burton/BritishMuseum-white/bm5-white_date_log/rose-blaze-27/checkpoint-5310",
"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": "LABEL_0"
},
"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": {
"LABEL_0": 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",
"squeeze_expansion_ratio": 0.25,
"strides": [
1,
2,
2,
2,
1,
2,
1
],
"torch_dtype": "float32",
"transformers_version": "4.38.2",
"width_coefficient": 1.2
}