{
    "experiment_name": "ss_c_base_no_excl",
    "experiment_group": "ss_mod",
    "lr": 0.0001,
    "weight_decay": 0.001,
    "val_freq": 1,
    "epochs": 30,
    "accumulation_steps": 16,
    "data": {
        "crop_size": null,
        "crop_type": null,
        "batch_size": 1,
        "feature_dir": "/mnt/disk2/domdet/features/nopae/2d_features_ss_mod",
        "label_dir": "/mnt/disk2/domdet/features/nopae/pairwise",
        "chains_csv": null,
        "evaluate_test": true,
        "eval_casp_10_plus": false,
        "splits_file": "splits_multi.json",
        "validation_splits": [
            "filt_val",
            "filt_val_multi",
            "filt_val_many"
        ],
        "remove_missing_residues": false,
        "using_alphafold_features": false,
        "recycling": true,
        "add_padding_mask": false,
        "training_exclusion_json": null,
        "multi_proportion": 0.65,
        "train_ids": null,
        "exclude_cath_test": false,
        "start_end_ss": false
    },
    "learner": {
        "uncertainty_model": false,
        "save_every_epoch": true,
        "model": {
            "type": "trrosetta",
            "kwargs": {
                "filters": 32,
                "kernel": 3,
                "num_layers": 31,
                "in_channels": 7,
                "symmetrise_output": true
            }
        },
        "assignment": {
            "type": "sparse_lowrank",
            "kwargs": {
                "N_iters": 3,
                "K_init": 4,
                "linker_threshold": 30
            }
        },
        "max_recycles": 1,
        "save_val_best": true,
        "x_has_padding_mask": false
    },
    "num_trainable_params": 577953
}