Update README.md
Browse files
README.md
CHANGED
|
@@ -27,4 +27,12 @@ from transformers import pipeline
|
|
| 27 |
pipe = pipeline("translation", model="sungmogi/en2ko_hiphop_small-100", src_lang="en", tgt_lang="ko")
|
| 28 |
|
| 29 |
pipe(input_text)
|
| 30 |
-
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
pipe = pipeline("translation", model="sungmogi/en2ko_hiphop_small-100", src_lang="en", tgt_lang="ko")
|
| 28 |
|
| 29 |
pipe(input_text)
|
| 30 |
+
```
|
| 31 |
+
|
| 32 |
+
# Training Hyperparameters
|
| 33 |
+
- per_device_train_batch_size: 4
|
| 34 |
+
- per_device_eval_batch_size: 4
|
| 35 |
+
- weight_decay: 0.01
|
| 36 |
+
- num_train_epochs: 4
|
| 37 |
+
- num_devices: 4
|
| 38 |
+
- learning_rate: 4e-5
|