examples/test-t5-custom-fields.okt

# 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/"
}