File size: 351 Bytes
8e55ba7 | 1 2 3 4 5 6 7 8 9 10 11 12 | {
"train_data": "data/train/predicate_detection_data/llm_annotated/llm_annotated_benchmark_stratified_filtered.jsonl",
"output_dir": "output/predicate_detector",
"model_name": "bert-base-uncased",
"val_ratio": 0.1,
"num_epochs": 10,
"batch_size": 16,
"learning_rate": 5e-05,
"seed": 42,
"max_seq_length": 256,
"debug_examples": 3
} |