禁止浪费！

train.py

@@ -8,8 +8,10 @@ from transformers import Trainer
 def train(env: Env) -> None:
     print(f"device = {env.device}")
 
-    train_data, _ = tools.preprocess_train_data(
+    train_data, test_data = tools.preprocess_train_data(
         "data/train.csv", env, upsample_factor=env.configs["upsample_factor"])
+    train_data = ds.concatenate_datasets([train_data, test_data])
+
     validate_data = tools.preprocess_test_data("data/dev.csv", env)
 
     trainer = Trainer(