File size: 543 Bytes
5df2c77 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
# okto_version: "1.2"
PROJECT "test_t5_basic"
DESCRIPTION "Teste basico com T5-small - sem blocos avancados"
ENV {
accelerator: "gpu"
min_memory: "4GB"
precision: "fp16"
backend: "oktoseek"
install_missing: true
}
DATASET {
train: "dataset/train.jsonl"
validation: "dataset/val.jsonl"
}
MODEL {
base: "google/t5-small"
device: "auto"
}
TRAIN {
epochs: 3
batch_size: 8
learning_rate: 0.0001
device: "auto"
}
EXPORT {
format: ["okm", "safetensors"]
path: "export/"
}
|