model_config.json

{
    "num_layers": 2,
    "dropout": 0.0,
    "hidden_dim": 512,
    "embedder_type": "MTEncoder",
    "input_dim": 512,
    "device": "cuda:0",
    "aggregation_mode": "mean"
}