File size: 547 Bytes
70f0877
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
{
  "train": "/home/diffusedit/data/split/train.jsonl",
  "val": "/home/diffusedit/data/split/val.jsonl",
  "test": "/home/diffusedit/data/split/test.jsonl",
  "out": "/home/diffusedit/full/runs/editor",
  "model": "google/flan-t5-large",
  "predict_only": false,
  "max_input": 1024,
  "max_new_tokens": 384,
  "lr": 0.0001,
  "batch": 4,
  "accum": 8,
  "epochs": 5,
  "patience": 1,
  "predict_batch": 8,
  "gradient_checkpointing": true,
  "log_every": 50,
  "limit": null,
  "seed": 42,
  "n_train": 12180,
  "n_val": 1564,
  "n_test": 1602
}