⚗️ Style Transfer: new hyperparameters in train.yaml

arkhn · Oct 16, 2024 · 74d82d7 · 74d82d7
1 parent 18ed9db
commit 74d82d7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/style-transfer/configs/rb_gen/train.yaml b/lib/style-transfer/configs/rb_gen/train.yaml
@@ -31,6 +31,9 @@ model:
 
 dataset:
   name: bio-datasets/mimic_style_transfer_tiny
+  sft_dataset:
+    name: bio-datasets/medical_transcription_for_seed
+    size: 10
   sft_ratio: 0.5
   gen_ratio: 0.5
   num_generated_samples: 6