Update README.md
Browse files
README.md
CHANGED
@@ -14,5 +14,29 @@ This is a Norwegian T5-base model trained on the Norwegian Colossal Corpus (NCC)
|
|
14 |
|
15 |
The following setting were used in training:
|
16 |
```bash
|
17 |
-
./run_t5_mlm_flax_streaming.py
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
18 |
```
|
|
|
14 |
|
15 |
The following setting were used in training:
|
16 |
```bash
|
17 |
+
./run_t5_mlm_flax_streaming.py \
|
18 |
+
--output_dir="./" \
|
19 |
+
--model_type="t5" \
|
20 |
+
--config_name="./" \
|
21 |
+
--tokenizer_name="./" \
|
22 |
+
--dataset_name="pere/norwegian_colossal_corpus_v2_short100k" \
|
23 |
+
--max_seq_length="512" \
|
24 |
+
--weight_decay="0.01" \
|
25 |
+
--per_device_train_batch_size="32" \
|
26 |
+
--per_device_eval_batch_size="32" \
|
27 |
+
--learning_rate="8e-3" \
|
28 |
+
--warmup_steps="0" \
|
29 |
+
--overwrite_output_dir \
|
30 |
+
--cache_dir /mnt/disks/flaxdisk/cache/ \
|
31 |
+
--num_train_epochs="5" \
|
32 |
+
--adam_beta1="0.9" \
|
33 |
+
--adam_beta2="0.98" \
|
34 |
+
--logging_steps="500" \
|
35 |
+
--num_train_steps="1000000" \
|
36 |
+
--num_eval_samples="5000" \
|
37 |
+
--save_steps="5000" \
|
38 |
+
--eval_steps="5000" \
|
39 |
+
--preprocessing_num_workers 96 \
|
40 |
+
--adafactor \
|
41 |
+
--push_to_hub
|
42 |
```
|