{ "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 }