pdazad's picture
Model save
7f1df03 verified
metadata
library_name: transformers
license: bigscience-bloom-rail-1.0
base_model: bigscience/bloom-560m
tags:
  - generated_from_trainer
model-index:
  - name: fine_tuned_bloom_owasp
    results: []

fine_tuned_bloom_owasp

This model is a fine-tuned version of bigscience/bloom-560m on the None dataset. It achieves the following results on the evaluation set:

  • Loss: 1.1134

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 5e-05
  • train_batch_size: 1
  • eval_batch_size: 1
  • seed: 42
  • gradient_accumulation_steps: 4
  • total_train_batch_size: 4
  • optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
  • lr_scheduler_type: linear
  • num_epochs: 8
  • mixed_precision_training: Native AMP

Training results

Training Loss Epoch Step Validation Loss
No log 1.0 14 1.5426
No log 2.0 28 0.9975
No log 3.0 42 0.8882
No log 4.0 56 0.9376
No log 5.0 70 0.9962
No log 6.0 84 1.1134

Framework versions

  • Transformers 4.46.3
  • Pytorch 2.5.1+cu121
  • Datasets 3.2.0
  • Tokenizers 0.20.3