diff --git a/.ipynb_checkpoints/config_poem-checkpoint.py b/.ipynb_checkpoints/config_poem-checkpoint.py
index bcb04e6..8792cbc 100644
--- a/.ipynb_checkpoints/config_poem-checkpoint.py
+++ b/.ipynb_checkpoints/config_poem-checkpoint.py
@@ -10,7 +10,7 @@
     #                     help='data file')
 
     parser.add_argument('--data_path', type=str,
-                        default='./datasets/yangsaisai-poetrydatasets-0_0_1/',
+                        default='./datasets/yangsaisai-poetrydatasets/',
                         help='data path')
 
 
@@ -107,7 +107,7 @@
             help='continue from the outputs in the given directory')
 
     logging_args.add_argument('--data_dir', type=str,
-            default='./datasets/yangsaisai-poetrydatasets-0_0_1',
+            default='./datasets/yangsaisai-poetrydatasets',
             help='data file path')
 
     logging_args.add_argument('--best_model', type=str,
diff --git a/config_poem.py b/config_poem.py
index bcb04e6..8792cbc 100644
--- a/config_poem.py
+++ b/config_poem.py
@@ -10,7 +10,7 @@
     #                     help='data file')
 
     parser.add_argument('--data_path', type=str,
-                        default='./datasets/yangsaisai-poetrydatasets-0_0_1/',
+                        default='./datasets/yangsaisai-poetrydatasets/',
                         help='data path')
 
 
@@ -107,7 +107,7 @@
             help='continue from the outputs in the given directory')
 
     logging_args.add_argument('--data_dir', type=str,
-            default='./datasets/yangsaisai-poetrydatasets-0_0_1',
+            default='./datasets/yangsaisai-poetrydatasets',
             help='data file path')
 
     logging_args.add_argument('--best_model', type=str,