Edit model card
YAML Metadata Warning: empty or missing yaml metadata in repo card (https://huggingface.co/docs/hub/model-cards#model-card-metadata)
  • Model: "functional_1"
    • Layer (type) ┃ Output Shape ┃ Param # ┃
    • input_layer (InputLayer) β”‚ (None, 256, 256, 3) β”‚ 0 β”‚
    • conv2d (Conv2D) β”‚ (None, 256, 256, 32) β”‚ 4,736 β”‚
    • conv2d_1 (Conv2D) β”‚ (None, 256, 256, 32) β”‚ 50,208 β”‚
    • max_pooling2d (MaxPooling2D) β”‚ (None, 64, 64, 32) β”‚ 0 β”‚
    • conv2d_2 (Conv2D) β”‚ (None, 64, 64, 64) β”‚ 51,264 β”‚
    • conv2d_3 (Conv2D) β”‚ (None, 64, 64, 64) β”‚ 102,464 β”‚
    • max_pooling2d_1 (MaxPooling2D) β”‚ (None, 32, 32, 64) β”‚ 0 β”‚
    • conv2d_4 (Conv2D) β”‚ (None, 32, 32, 128) β”‚ 73,856 β”‚
    • conv2d_5 (Conv2D) β”‚ (None, 32, 32, 128) β”‚ 147,584 β”‚
    • max_pooling2d_2 (MaxPooling2D) β”‚ (None, 16, 16, 128) β”‚ 0 β”‚
    • conv2d_6 (Conv2D) β”‚ (None, 16, 16, 256) β”‚ 295,168 β”‚
    • conv2d_7 (Conv2D) β”‚ (None, 16, 16, 256) β”‚ 590,080 β”‚
    • max_pooling2d_3 (MaxPooling2D) β”‚ (None, 8, 8, 256) β”‚ 0 β”‚
    • flatten (Flatten) β”‚ (None, 16384) β”‚ 0 β”‚
    • dense (Dense) β”‚ (None, 1024) β”‚ 16,778,240 β”‚
    • dense_1 (Dense) β”‚ (None, 512) β”‚ 524,800 β”‚
    • dense_2 (Dense) β”‚ (None, 256) β”‚ 131,328 β”‚
    • dense_3 (Dense) β”‚ (None, 128) β”‚ 32,896 β”‚
    • dense_4 (Dense) β”‚ (None, 8) β”‚ 1,032 β”‚
  • Total params: 56,350,970 (214.96 MB)
  • Trainable params: 18,783,656 (71.65 MB)
  • Non-trainable params: 0 (0.00 B)
  • Optimizer params: 37,567,314 (143.31 MB)

Test Loss: 0.001258564880117774, Test MAE: 0.018795978277921677

Downloads last month
3
Inference API
Unable to determine this model’s pipeline type. Check the docs .