oktoscript / examples /test-t5-custom-fields.okt
OktoSeek's picture
Update
5df2c77 verified
# okto_version: "1.2"
PROJECT "test_t5_custom_fields"
DESCRIPTION "Exemplo usando campos customizados no dataset (input_field e output_field)"
ENV {
accelerator: "gpu"
min_memory: "4GB"
precision: "fp16"
backend: "oktoseek"
install_missing: true
}
DATASET {
train: "dataset/train.jsonl"
validation: "dataset/val.jsonl"
format: "jsonl"
type: "chat"
# Campos customizados: define quais colunas usar do dataset
input_field: "input"
output_field: "target"
}
MODEL {
base: "t5-small"
device: "auto"
}
TRAIN {
epochs: 3
batch_size: 8
learning_rate: 0.0001
device: "auto"
}
EXPORT {
format: ["okm", "safetensors"]
path: "export/"
}