英文

arzwiki_mlm(arzRoBERTa)

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

  • 伪困惑度:

模型描述

需要更多信息

预期用途和限制

需要更多信息

训练和评估数据

需要更多信息

训练过程

训练超参数

训练过程中使用了以下超参数:

  • learning_rate: 0.0001
  • train_batch_size: 256
  • 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 2500 2.038300
2 5000 0.878800
3 7500 0.682800
4 10000 0.613100
5 12500 0.574500
Train Runtime Train Samples Per Second Train Steps Per Second Total Flos Train Loss Epoch
14677.117400 248.119000 0.970000 120746231839334400.000000 0.908513 5.000000

框架版本

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