examples/basic-training/scripts/train.okt

# okto_version: "1.0"

PROJECT "BasicModel"
DESCRIPTION "Basic training example for OktoEngine"

ENV {

  accelerator: "gpu"

  min_memory: "8GB"

  precision: "fp16"

  backend: "oktoseek"

  install_missing: true
}

DATASET {

  train: "dataset/train.jsonl"

  validation: "dataset/val.jsonl"
}

MODEL {

  base: "gpt2"
}

TRAIN {

  epochs: 5

  batch_size: 32

  device: "auto"
}

EXPORT {

  format: ["okm"]

  path: "export/"
}