config.json

{
 "_name_or_path": "microsoft/resnet-50",
 "_num_labels": 5,
 "architectures": [
 "ResNetForImageClassification"
 ],
 "depths": [
 3,
 4,
 6,
 3
 ],
 "downsample_in_bottleneck": false,
 "downsample_in_first_stage": false,
 "embedding_size": 64,
 "hidden_act": "relu",
 "hidden_sizes": [
 256,
 512,
 1024,
 2048
 ],
 "id2label": {
 "0": "10_just_a_pure_cotton_quilt",
 "1": "11_cover_the_quilt_with_only_the_head_exposed",
 "2": "12_just_a_pure_face",
 "3": "13_cover_the_quilt_to_expose_the_head_and_shoulders",
 "4": "14_has_nothing_to_do_with_11_and_13_above"
 },
 "label2id": {
 "10_just_a_pure_cotton_quilt": 0,
 "11_cover_the_quilt_with_only_the_head_exposed": 1,
 "12_just_a_pure_face": 2,
 "13_cover_the_quilt_to_expose_the_head_and_shoulders": 3,
 "14_has_nothing_to_do_with_11_and_13_above": 4
 },
 "layer_type": "bottleneck",
 "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.36.1"
}