{
    "docstring": "SeisT-S (Li et al., 2024, https://doi.org/10.1109/TGRS.2024.3371503) for earthquake detection and P/S phase picking, trained on the training splits of STEAD, INSTANCE (events and noise), CREW and CEED at 100 Hz (6.1 million traces, 100 million training windows of 60 s with picks at random positions). Not the weights of the original publication, which were trained on DiTing at 50 Hz.\nThresholds selected for optimal F1 score on the pooled development splits of the training datasets. Depending on the target region, the thresholds might need to be adjusted.\nWhen using this model, please reference Li et al. (2024) and the SeisBench publications listed at https://github.com/seisbench/seisbench\n\nSen Li (senli.1073@gmail.com)",
    "model_args": {
        "sampling_rate": 100,
        "component_order": "ZNE",
        "in_samples": 6000,
        "filter_args": null,
        "filter_kwargs": null,
        "grouping": "instrument",
        "in_channels": 3,
        "classes": 2,
        "phases": "PS",
        "variant": "s",
        "norm": "std",
        "stem_channels": [
            16,
            8,
            16,
            16
        ],
        "stem_kernel_sizes": [
            11,
            5,
            5,
            7
        ],
        "stem_strides": [
            2,
            1,
            1,
            2
        ],
        "layer_blocks": [
            2,
            2,
            3,
            2
        ],
        "layer_channels": [
            16,
            24,
            32,
            64
        ],
        "attn_blocks": [
            1,
            1,
            1,
            1
        ],
        "stage_aggr_ratios": [
            2,
            2,
            2,
            2
        ],
        "attn_aggr_ratios": [
            8,
            4,
            2,
            1
        ],
        "head_dims": [
            8,
            8,
            8,
            16
        ],
        "msmc_kernel_sizes": [
            5,
            7
        ],
        "path_drop_rate": 0.1,
        "attn_drop_rate": 0.1,
        "key_drop_rate": 0.1,
        "mlp_drop_rate": 0.1,
        "other_drop_rate": 0.1,
        "attn_ratio": 0.6,
        "mlp_ratio": 2,
        "qkv_bias": true,
        "mlp_bias": true
    },
    "seisbench_requirement": "0.13.0",
    "version": "1",
    "default_args": {
        "P_threshold": 0.3,
        "S_threshold": 0.25,
        "detection_threshold": 0.45
    }
}