From 7c418cf2d8dea1a959f77e7231d1f07bcc86ed74 Mon Sep 17 00:00:00 2001
From: xingzhaohu <920232796@qq.com>
Date: Tue, 20 Apr 2021 16:15:35 +0800
Subject: [PATCH] ancient translation test

---
 .DS_Store                                  | Bin 18436 -> 18436 bytes
 examples/gpt2_ancient_translation_train.py |   2 +-
 test/gpt_ancient_translation_test.py       |   2 +-
 3 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/.DS_Store b/.DS_Store
index 3ebbcefc5239b69bcd2ca269d510d63939616090..5600a6f0d7cf170d7913de2c488bccc1e8c998ce 100644
GIT binary patch
delta 27
ecmZpfz}PZ@af2=U<jn$}o7>p8YeN{ubs_+h{|bEo

delta 27
ecmZpfz}PZ@af2=U<a%+>&28-4wIK}SIuQVmj0z(F

diff --git a/examples/gpt2_ancient_translation_train.py b/examples/gpt2_ancient_translation_train.py
index 0ea4a21..66897a2 100644
--- a/examples/gpt2_ancient_translation_train.py
+++ b/examples/gpt2_ancient_translation_train.py
@@ -187,7 +187,7 @@ def iteration(self, epoch, dataloader, train=True):
 if __name__ == '__main__':
 
     trainer = Trainer()
-    train_epoches = 10
+    train_epoches = 100
     for epoch in range(train_epoches):
         # 训练一个epoch
         trainer.train(epoch)
diff --git a/test/gpt_ancient_translation_test.py b/test/gpt_ancient_translation_test.py
index 70850df..dba05db 100644
--- a/test/gpt_ancient_translation_test.py
+++ b/test/gpt_ancient_translation_test.py
@@ -17,4 +17,4 @@
     model.set_device(device)
     model.load_all_params(model_path)
 
-    print(model.sample_generate("自昔羲后，因以物命官，事简人淳，唯以道化，上无求欲於下，下无干进於上，百姓自足，海内乂安，不是贤而非愚，不沽名而尚行，推择之典，无所闻焉。", out_max_length=300, add_eos=True))
\ No newline at end of file
+    print(model.sample_generate("余忆童稚时，能张目对日。", out_max_length=300, add_eos=True))
\ No newline at end of file