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