ITT-AF's picture
Update README.md
af32456 verified
---
license: cc-by-nc-4.0
---
# ITT-AF/ITT-AF-PLM-2.2B_v0.2
This model is a pretrained version on custom dataset(110G).
## Model description
More information needed
## Intended uses & limitations
More information needed
## Training and evaluation data
The data used to train the model is collected from various sources, mostly from the Web.
As such, it contains offensive, harmful and biased content.
We thus expect the model to exhibit such biases from the training data.
## Training procedure
### Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 2e-05
- train_batch_size: 24
- eval_batch_size: 8
- seed: 42
- gradient_accumulation_steps: 4
- total_train_batch_size: 96
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: linear
- num_epochs: 1.0
- mixed_precision_training: Native AMP
### Training results
```
text = "ν•œκ΅­μ˜ μˆ˜λ„λŠ”"
gen_text = "ν•œκ΅­μ˜ μˆ˜λ„λŠ” μ„œμšΈμ΄λ‹€. κ·ΈλŸ¬λ‚˜ μ„œμšΈμ΄λΌλŠ” λ„μ‹œλŠ” κ·Έ μžμ²΄κ°€ ν•˜λ‚˜μ˜ κ±°λŒ€ν•œ λ„μ‹œλ‹€. μ„œμšΈμ˜ 쀑심은 κ΄‘ν™”λ¬Έκ΄‘μž₯이닀."
```
### Framework versions
- Transformers 4.36.2
- Pytorch 2.1.2+cu121
- Datasets 2.0.0
- Tokenizers 0.15.0