| { | |
| "model_type": "resnet", | |
| "input_shape": [224, 224, 3], | |
| "num_classes": 2, | |
| "layers": [ | |
| {"type": "resnet152", "output_shape": [7, 7, 2048]}, | |
| {"type": "global_average_pooling2d", "output_shape": [2048]}, | |
| {"type": "dense", "output_shape": [1]} | |
| ], | |
| "total_parameters": 58377093 | |
| } |