英文

arywiki_mlm (aryRoBERTa)

该模型是在未知数据集上基于 进行微调的版本。在评估集上取得以下结果:

  • 伪困惑度:

模型描述

需要更多信息

预期用途和限制

需要更多信息

训练和评估数据

需要更多信息

训练过程

训练超参数

训练时使用了以下超参数:

  • learning_rate: 0.0001
  • train_batch_size: 128
  • eval_batch_size: 8
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.98) and epsilon=1e-06
  • lr_scheduler_type: linear
  • num_epochs: 5

训练结果

Epoch Step Training Loss
1 35 9.561500
2 70 7.946000
3 105 7.420400
4 140 7.197800
5 175 7.174400
Train Runtime Train Samples Per Second Train Steps Per Second Total Flos Train Loss Epoch
192.684800 121.260000 0.960000 774708261150720.000000 7.812142 5.000000

框架版本

  • Datasets 2.9.0
  • Tokenizers 0.12.1
  • Transformers 4.24.0
  • Pytorch 1.12.1+cu116