schnapp's picture
End of training
304ad1e verified
|
raw
history blame
2.96 kB
metadata
library_name: peft
tags:
  - generated_from_trainer
base_model: microsoft/codebert-base-mlm
model-index:
  - name: malicious-package-classifier-bert-mal-only
    results: []

malicious-package-classifier-bert-mal-only

This model is a fine-tuned version of microsoft/codebert-base-mlm on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 0.0000

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: 140
  • eval_batch_size: 140
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 32

Training results

Training Loss Epoch Step Validation Loss
No log 1.0 42 0.0129
No log 2.0 84 0.0014
No log 3.0 126 0.0007
No log 4.0 168 0.0005
No log 5.0 210 0.0003
No log 6.0 252 0.0003
No log 7.0 294 0.0002
No log 8.0 336 0.0002
No log 9.0 378 0.0001
No log 10.0 420 0.0001
No log 11.0 462 0.0001
0.0161 12.0 504 0.0001
0.0161 13.0 546 0.0001
0.0161 14.0 588 0.0001
0.0161 15.0 630 0.0001
0.0161 16.0 672 0.0001
0.0161 17.0 714 0.0001
0.0161 18.0 756 0.0001
0.0161 19.0 798 0.0000
0.0161 20.0 840 0.0000
0.0161 21.0 882 0.0000
0.0161 22.0 924 0.0000
0.0161 23.0 966 0.0000
0.0 24.0 1008 0.0000
0.0 25.0 1050 0.0000
0.0 26.0 1092 0.0000
0.0 27.0 1134 0.0000
0.0 28.0 1176 0.0000
0.0 29.0 1218 0.0000
0.0 30.0 1260 0.0000
0.0 31.0 1302 0.0000
0.0 32.0 1344 0.0000

Framework versions

  • PEFT 0.10.0
  • Transformers 4.39.3
  • Pytorch 2.2.2+cu121
  • Datasets 2.18.0
  • Tokenizers 0.15.2