params.json

{
  "dataset_args": {
    "history_max_utterances": 1000000,
    "history_max_tokens": 128,
    "dataroot": "data_exclude_kb",
    "knowledge_file": "knowledge.json"
  },
  "task": "clusterid_classification",
  "model_name_or_path": "bert-base-uncased",
  "tokenizer_name_or_path": "/scratch/yt2267/public/alexa-with-dstc9-track1-dataset/bert-base-uncased-local",
  "per_gpu_train_batch_size": 16,
  "per_gpu_eval_batch_size": 32,
  "gradient_accumulation_steps": 1,
  "learning_rate": 6.25e-05,
  "adam_epsilon": 1e-08,
  "max_grad_norm": 1.0,
  "num_train_epochs": 100,
  "warmup_steps": 0,
  "fp16": "",
  "seed": 42
}