OwusuBlessing
30 lines
450 B
YAML
30 lines
450 B
YAML
task:
|
|
name: "styling"
|
|
type: "style_transfer"
|
|
|
|
data:
|
|
source: "custom"
|
|
input_field: "text"
|
|
style_field: "style"
|
|
max_length: 256
|
|
train_split: 0.8
|
|
validation_split: 0.1
|
|
test_split: 0.1
|
|
|
|
model:
|
|
name: "t5-base"
|
|
max_length: 256
|
|
|
|
training:
|
|
num_epochs: 3
|
|
batch_size: 16
|
|
learning_rate: 3e-5
|
|
weight_decay: 0.01
|
|
warmup_ratio: 0.1
|
|
lr_scheduler_type: "linear"
|
|
|
|
inference:
|
|
batch_size: 32
|
|
max_new_tokens: 128
|
|
temperature: 0.8
|