add seq2seq configs

huggingface · Aug 14, 2024 · b731e72 · b731e72
1 parent 3cb6510
commit b731e72
Show file tree

Hide file tree

Showing 2 changed files with 57 additions and 0 deletions.
diff --git a/configs/seq2seq/hub_dataset.yml b/configs/seq2seq/hub_dataset.yml
@@ -0,0 +1,28 @@
+task: seq2seq
+base_model: google/flan-t5-base
+project_name: autotrain-seq2seq-hub-dataset
+log: tensorboard
+backend: local
+
+data:
+  path: samsum
+  train_split: train
+  valid_split: test
+  column_mapping:
+    text_column: dialogue
+    target_column: summary
+
+params:
+  max_seq_length: 512
+  epochs: 3
+  batch_size: 4
+  lr: 2e-5
+  optimizer: adamw_torch
+  scheduler: linear
+  gradient_accumulation: 1
+  mixed_precision: none
+
+hub:
+  username: ${HF_USERNAME}
+  token: ${HF_TOKEN}
+  push_to_hub: true
diff --git a/configs/seq2seq/local.yml b/configs/seq2seq/local.yml
@@ -0,0 +1,29 @@
+task: seq2seq
+base_model: google/flan-t5-base
+project_name: autotrain-seq2seq-local
+log: tensorboard
+backend: local
+
+data:
+  path: path/to/your/dataset csv/jsonl files
+  train_split: train
+  valid_split: test
+  column_mapping:
+    text_column: text
+    target_column: target
+
+
+params:
+  max_seq_length: 512
+  epochs: 3
+  batch_size: 4
+  lr: 2e-5
+  optimizer: adamw_torch
+  scheduler: linear
+  gradient_accumulation: 1
+  mixed_precision: none
+
+hub:
+  username: ${HF_USERNAME}
+  token: ${HF_TOKEN}
+  push_to_hub: true