ancient translation test

xyyhcl · Apr 20, 2021 · 7c418cf · 7c418cf
1 parent a933c37
commit 7c418cf
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/.DS_Store b/.DS_Store
diff --git 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
@@ -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))
+    print(model.sample_generate("余忆童稚时，能张目对日。", out_max_length=300, add_eos=True))