model_params.json

{
    "model": {
                "cnn":{
                    "model_name": "cnn",
                    "out_channels": [16, 32, 32],
                    "pool_size": [2, 1],
                    "strides": [2, 1],
                    "filter_sizes": [[9, 1], [9, 1], [9, 1]],
                    "fc": [500, 200],
                    "drop_rates": [0.5, 0.5, 0.5],
                    "fc_drop_rates": [0.5, 0.5]

                },ß
                "cnn_resi_ecg":{
                    "model_name": "cnn_resi_ecg",
                    "out_channel": 16,
                    "num_layers_in_res": 2,
                    "num_res_blocks": 15,
                    "pool_size": [2, 1],
                    "stride": [2, 1],
                    "filter_size": [32, 1],
                    "fc": [128],
                    "drop_rate": 0.2,
                    "fc_drop_rates": [0.5]

                },
                "cnn_resi_cam_ecg":{
                    "model_name": "cnn_resi_cam_ecg",
                    "out_channel": 16,
                    "num_layers_in_res": 2,
                    "num_res_blocks": 15,
                    "pool_size": [2, 1],
                    "stride": [2, 1],
                    "filter_size": [32, 1],
                    "drop_rate": 0.2

                },
                "fnn":{
                    "model_name": "fnn",
                    "layer_dims": [1024, 256, 128],
                    "droprate": 0.5

                }
    }
}