nimble-bert-v2 / training_args.json
Bakobiibizo's picture
Upload folder using huggingface_hub
e302db6 verified
raw
history blame contribute delete
No virus
1.15 kB
{
"model_name_or_path": "lstm",
"attack": null,
"dataset": "yelp_polarity",
"task_type": "classification",
"model_max_length": null,
"model_num_labels": null,
"dataset_train_split": "train",
"dataset_eval_split": "test",
"filter_train_by_labels": null,
"filter_eval_by_labels": null,
"num_epochs": 50,
"num_clean_epochs": 1,
"attack_epoch_interval": 1,
"early_stopping_epochs": null,
"learning_rate": 1e-05,
"num_warmup_steps": 500,
"weight_decay": 0.01,
"per_device_train_batch_size": 8,
"per_device_eval_batch_size": 32,
"gradient_accumulation_steps": 1,
"random_seed": 786,
"parallel": false,
"load_best_model_at_end": false,
"alpha": 1.0,
"num_train_adv_examples": -1,
"query_budget_train": null,
"attack_num_workers_per_device": 1,
"output_dir": "./outputs/2024-03-22-01-16-17-693140",
"checkpoint_interval_steps": null,
"checkpoint_interval_epochs": null,
"save_last": true,
"log_to_tb": false,
"tb_log_dir": null,
"log_to_wandb": false,
"wandb_project": "textattack",
"logging_interval_step": 1
}