training_config.json

{
  "base_model": "zilliz/semantic-highlight-bilingual-v1",
  "max_length": 4096,
  "num_labels": 2,
  "ranking_weight": 0.0,
  "pruning_weight": 1.0,
  "learning_rate": 5e-05,
  "batch_size": 4,
  "num_epochs": 2,
  "world_size": 1
}