train_args.json

{
  "model": "bert-base-uncased",
  "dataset": "snli",
  "num_labels": 3
}