{
  "model_save_dir": "models",
  "model_save_name": "linkage_multi_aliases",
  "opt_model_description": "This model was trained on a dataset consisting of company aliases from wiki data using the LinkTransformer framework. \n                               It was trained for 70 epochs using other defaults that can be found in the repo's LinkTransformer config file - LT_training_config.json \n  ",
  "opt_model_lang": [
    "de",
    "en",
    "zh",
    "ja",
    "hi",
    "ar",
    "bn",
    "pt",
    "ru",
    "es",
    "fr",
    "ko"
  ],
  "train_batch_size": 64,
  "num_epochs": 70,
  "warm_up_perc": 1,
  "learning_rate": 2e-05,
  "loss_type": "supcon",
  "val_perc": 0.2,
  "wandb_names": {
    "project": "linkage",
    "id": "econabhishek",
    "run": "linkage_multi_aliases",
    "entity": "econabhishek"
  },
  "add_pooling_layer": false,
  "large_val": true,
  "eval_steps_perc": 0.5,
  "test_at_end": true,
  "save_val_test_pickles": true,
  "val_query_prop": 0.5,
  "loss_params": {},
  "eval_type": "retrieval",
  "training_dataset": "dataframe",
  "base_model_path": "sentence-transformers/paraphrase-multilingual-mpnet-base-v2",
  "best_model_path": "models/linkage_multi_aliases"
}