Spaces:
Sleeping
Sleeping
| { | |
| "training_args": { | |
| "output_dir": "./output", | |
| "overwrite_output_dir": true, | |
| "num_train_epochs": 3, | |
| "per_device_train_batch_size": 16, | |
| "per_device_eval_batch_size": 64, | |
| "save_steps": 10000, | |
| "save_total_limit": 1 | |
| }, | |
| "model_args": { | |
| "num_labels": 2, | |
| "hidden_dropout_prob": 0.1, | |
| "hidden_size": 768, | |
| "num_hidden_layers": 12, | |
| "num_attention_heads": 12, | |
| "max_position_embeddings": 512 | |
| }, | |
| "data_args": { | |
| "train_file": "./data/train.csv", | |
| "test_file": "./data/test.csv", | |
| "validation_file": "./data/val.csv" | |
| }, | |
| "tokenizer_name": "TunedModel" | |
| } | |