Training in progress, step 500
Browse files- aT5translation.py +1 -1
- training_args.bin +1 -1
aT5translation.py
CHANGED
|
@@ -47,7 +47,7 @@ tokenized_datasets = raw_datasets.map(preprocess_function, batched=True)
|
|
| 47 |
from transformers import AutoModelForSeq2SeqLM, DataCollatorForSeq2Seq, Seq2SeqTrainingArguments, Seq2SeqTrainer
|
| 48 |
model=MT5ForConditionalGeneration.from_pretrained(path)
|
| 49 |
#各変数の定義
|
| 50 |
-
batch_size =
|
| 51 |
|
| 52 |
print("args")
|
| 53 |
args = Seq2SeqTrainingArguments(
|
|
|
|
| 47 |
from transformers import AutoModelForSeq2SeqLM, DataCollatorForSeq2Seq, Seq2SeqTrainingArguments, Seq2SeqTrainer
|
| 48 |
model=MT5ForConditionalGeneration.from_pretrained(path)
|
| 49 |
#各変数の定義
|
| 50 |
+
batch_size = 8
|
| 51 |
|
| 52 |
print("args")
|
| 53 |
args = Seq2SeqTrainingArguments(
|
training_args.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 4832
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7e069a95fe3ac615057547bdbfdff581966bbb1e2f2d42648587283ac5bb6579
|
| 3 |
size 4832
|