config.json 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126
  1. {
  2. "_name_or_path": "/dccstor/doc_fig_class/figure_classifier/final_model",
  3. "architectures": [
  4. "EfficientNetForImageClassification"
  5. ],
  6. "batch_norm_eps": 0.001,
  7. "batch_norm_momentum": 0.99,
  8. "depth_coefficient": 1.0,
  9. "depth_divisor": 8,
  10. "depthwise_padding": [],
  11. "drop_connect_rate": 0.2,
  12. "dropout_rate": 0.2,
  13. "expand_ratios": [
  14. 1,
  15. 6,
  16. 6,
  17. 6,
  18. 6,
  19. 6,
  20. 6
  21. ],
  22. "hidden_act": "swish",
  23. "hidden_dim": 1280,
  24. "id2label": {
  25. "0": "bar_chart",
  26. "1": "bar_code",
  27. "2": "chemistry_markush_structure",
  28. "3": "chemistry_molecular_structure",
  29. "4": "flow_chart",
  30. "5": "icon",
  31. "6": "line_chart",
  32. "7": "logo",
  33. "8": "map",
  34. "9": "other",
  35. "10": "pie_chart",
  36. "11": "qr_code",
  37. "12": "remote_sensing",
  38. "13": "screenshot",
  39. "14": "signature",
  40. "15": "stamp"
  41. },
  42. "image_size": 224,
  43. "in_channels": [
  44. 32,
  45. 16,
  46. 24,
  47. 40,
  48. 80,
  49. 112,
  50. 192
  51. ],
  52. "initializer_range": 0.02,
  53. "kernel_sizes": [
  54. 3,
  55. 3,
  56. 5,
  57. 3,
  58. 5,
  59. 5,
  60. 3
  61. ],
  62. "label2id": {
  63. "bar_chart": 0,
  64. "bar_code": 1,
  65. "chemistry_markush_structure": 2,
  66. "chemistry_molecular_structure": 3,
  67. "flow_chart": 4,
  68. "icon": 5,
  69. "line_chart": 6,
  70. "logo": 7,
  71. "map": 8,
  72. "other": 9,
  73. "pie_chart": 10,
  74. "qr_code": 11,
  75. "remote_sensing": 12,
  76. "screenshot": 13,
  77. "signature": 14,
  78. "stamp": 15
  79. },
  80. "model_type": "efficientnet",
  81. "num_block_repeats": [
  82. 1,
  83. 2,
  84. 2,
  85. 3,
  86. 3,
  87. 4,
  88. 1
  89. ],
  90. "num_channels": 3,
  91. "num_hidden_layers": 64,
  92. "out_channels": [
  93. 16,
  94. 24,
  95. 40,
  96. 80,
  97. 112,
  98. 192,
  99. 320
  100. ],
  101. "out_features": null,
  102. "pooling_type": "mean",
  103. "squeeze_expansion_ratio": 0.25,
  104. "stage_names": [
  105. "stem",
  106. "stage1",
  107. "stage2",
  108. "stage3",
  109. "stage4",
  110. "stage5",
  111. "stage6",
  112. "stage7"
  113. ],
  114. "strides": [
  115. 1,
  116. 2,
  117. 2,
  118. 2,
  119. 1,
  120. 2,
  121. 1
  122. ],
  123. "torch_dtype": "float32",
  124. "transformers_version": "4.48.0",
  125. "width_coefficient": 1.0
  126. }