🐛 Style Transfer: tweak dataset split proportion

arkhn · Nov 14, 2024 · d61dfa9 · d61dfa9
1 parent 85c7c74
commit d61dfa9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/style-transfer/style_transfer/rb_gen/steps/sft.py b/lib/style-transfer/style_transfer/rb_gen/steps/sft.py
@@ -64,7 +64,7 @@ def sft_train(
     test_sft_dataset = None
     if cfg.dataset.sft_dataset is not None:
         sft_dataset, test_sft_dataset = sft_dataset.train_test_split(
-            train_size=0.1, shuffle=False
+            train_size=0.5, shuffle=False
         ).values()
     args.load_best_model_at_end = True
     trainer = SFTTrainer(