config_chem_mrl.json

{
    "__version__": "0.7.2",
    "embedding_pooling": "mean",
    "eval_metric": "spearman",
    "eval_similarity_fct": "tanimoto",
    "kl_div_weight": 0.5,
    "kl_temperature": 0.3,
    "last_layer_weight": 2.0,
    "loss_func": "tanimotosentloss",
    "model_name": "Derify/ChemBERTa-druglike",
    "mrl_dimension_weights": [
        1,
        1,
        1,
        1,
        1,
        1,
        1,
        1
    ],
    "mrl_dimensions": [
        1024,
        512,
        256,
        128,
        64,
        32,
        16,
        8
    ],
    "n_dims_per_step": -1,
    "n_layers_per_step": -1,
    "prior_layers_weight": 1.0,
    "tanimoto_similarity_loss_func": null,
    "use_2d_matryoshka": true,
    "use_query_tokenizer": false
}