training_config.json

{
  "batch": 32,
  "lr": 2e-05,
  "epochs": 20,
  "dataset_path": "/kaggle/working/finetuning_framework_for_cyberbullying/data/1_Multilablel_Cyberbully_Data.csv",
  "model_path": "sagorsarker/bangla-bert-base",
  "max_length": 128,
  "num_folds": 5,
  "freeze_base": false,
  "seed": 42,
  "stratification_type": "multilabel",
  "author_name": "saif_siddique",
  "mlflow_experiment_name": "Bangla-Cyberbullying-Detection",
  "dropout": 0.1,
  "weight_decay": 0.01,
  "warmup_ratio": 0.1,
  "gradient_clip_norm": 1.0,
  "early_stopping_patience": 5,
  "no_amp": false,
  "no_cache": false,
  "save_model_dir": "./saved_models",
  "no_save_model": false,
  "push_to_hub": true,
  "hub_repo_name": "Saif-Siddique/bangla-cyberbully-sagor-bangla-bert-base",
  "hub_private": false,
  "output_dir": "./outputs",
  "cache_dir": "./cache"
}