diff --git a/ft.py b/ft.py
index 1a7a8fe..6fc32c2 100644
--- a/ft.py
+++ b/ft.py
@@ -129,10 +129,10 @@ class FT:
 
     def train(self):
         train_dataloader, test_dataloader = load_dataset(
-            "./dataset/hason_out.json",
+            "./dataset/hason_three.json",
             200,
             200,
-            1,
+            3,
             test_ratio=0.2,
         )
         num_epochs = 100