CoolFace
Modelpublic

PL-RnD/privacy-moderation-small

sourceHugging Faceapache-2.0updated 1y agoView on Hugging Face
1likes35downloads
training_args.json145 linesDownload Raw Back to root
1{2  "output_dir": "out",3  "overwrite_output_dir": false,4  "do_train": false,5  "do_eval": true,6  "do_predict": false,7  "eval_strategy": "epoch",8  "prediction_loss_only": false,9  "per_device_train_batch_size": 16,10  "per_device_eval_batch_size": 32,11  "per_gpu_train_batch_size": null,12  "per_gpu_eval_batch_size": null,13  "gradient_accumulation_steps": 16,14  "eval_accumulation_steps": null,15  "eval_delay": 0,16  "torch_empty_cache_steps": null,17  "learning_rate": 2e-05,18  "weight_decay": 0.01,19  "adam_beta1": 0.9,20  "adam_beta2": 0.999,21  "adam_epsilon": 1e-08,22  "max_grad_norm": 1.0,23  "num_train_epochs": 4.0,24  "max_steps": -1,25  "lr_scheduler_type": "linear",26  "lr_scheduler_kwargs": {},27  "warmup_ratio": 0.0,28  "warmup_steps": 400,29  "log_level": "passive",30  "log_level_replica": "warning",31  "log_on_each_node": true,32  "logging_dir": "out\\runs\\Sep14_18-56-43_DESKTOP-BHJ1RIF",33  "logging_strategy": "steps",34  "logging_first_step": false,35  "logging_steps": 50,36  "logging_nan_inf_filter": true,37  "save_strategy": "epoch",38  "save_steps": 500,39  "save_total_limit": null,40  "save_safetensors": true,41  "save_on_each_node": false,42  "save_only_model": false,43  "restore_callback_states_from_checkpoint": false,44  "no_cuda": false,45  "use_cpu": false,46  "use_mps_device": false,47  "seed": 42,48  "data_seed": null,49  "jit_mode_eval": false,50  "use_ipex": false,51  "bf16": true,52  "fp16": false,53  "fp16_opt_level": "O1",54  "half_precision_backend": "auto",55  "bf16_full_eval": true,56  "fp16_full_eval": false,57  "tf32": null,58  "local_rank": 0,59  "ddp_backend": null,60  "tpu_num_cores": null,61  "tpu_metrics_debug": false,62  "debug": [],63  "dataloader_drop_last": false,64  "eval_steps": null,65  "dataloader_num_workers": 2,66  "dataloader_prefetch_factor": null,67  "past_index": -1,68  "run_name": null,69  "disable_tqdm": false,70  "remove_unused_columns": true,71  "label_names": null,72  "load_best_model_at_end": true,73  "metric_for_best_model": "f1",74  "greater_is_better": true,75  "ignore_data_skip": false,76  "fsdp": [],77  "fsdp_min_num_params": 0,78  "fsdp_config": {79    "min_num_params": 0,80    "xla": false,81    "xla_fsdp_v2": false,82    "xla_fsdp_grad_ckpt": false83  },84  "fsdp_transformer_layer_cls_to_wrap": null,85  "accelerator_config": {86    "split_batches": false,87    "dispatch_batches": null,88    "even_batches": true,89    "use_seedable_sampler": true,90    "non_blocking": false,91    "gradient_accumulation_kwargs": null92  },93  "parallelism_config": null,94  "deepspeed": null,95  "label_smoothing_factor": 0.0,96  "optim": "adamw_torch_fused",97  "optim_args": null,98  "adafactor": false,99  "group_by_length": false,100  "length_column_name": "length",101  "report_to": [],102  "ddp_find_unused_parameters": null,103  "ddp_bucket_cap_mb": null,104  "ddp_broadcast_buffers": null,105  "dataloader_pin_memory": true,106  "dataloader_persistent_workers": false,107  "skip_memory_metrics": true,108  "use_legacy_prediction_loop": false,109  "push_to_hub": false,110  "resume_from_checkpoint": null,111  "hub_model_id": null,112  "hub_strategy": "every_save",113  "hub_token": "<HUB_TOKEN>",114  "hub_private_repo": null,115  "hub_always_push": false,116  "hub_revision": null,117  "gradient_checkpointing": false,118  "gradient_checkpointing_kwargs": null,119  "include_inputs_for_metrics": false,120  "include_for_metrics": [],121  "eval_do_concat_batches": true,122  "fp16_backend": "auto",123  "push_to_hub_model_id": null,124  "push_to_hub_organization": null,125  "push_to_hub_token": "<PUSH_TO_HUB_TOKEN>",126  "mp_parameters": "",127  "auto_find_batch_size": false,128  "full_determinism": false,129  "torchdynamo": null,130  "ray_scope": "last",131  "ddp_timeout": 1800,132  "torch_compile": false,133  "torch_compile_backend": null,134  "torch_compile_mode": null,135  "include_tokens_per_second": false,136  "include_num_input_tokens_seen": false,137  "neftune_noise_alpha": null,138  "optim_target_modules": null,139  "batch_eval_metrics": false,140  "eval_on_start": false,141  "use_liger_kernel": false,142  "liger_kernel_config": null,143  "eval_use_gather_object": false,144  "average_tokens_across_devices": false145}