File size: 1,948 Bytes
b1baf92
9a77f94
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1a86e6c
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

- 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