|
|
# okto_version: "1.2" |
|
|
|
|
|
# Teste 6: T5 com EXPLORER - AutoML Básico |
|
|
# Modelo: google/t5-small |
|
|
# Objetivo: Testar bloco EXPLORER para busca de hiperparâmetros |
|
|
|
|
|
PROJECT "test_t5_explorer" |
|
|
DESCRIPTION "Teste T5 com EXPLORER - busca automática de hiperparâmetros" |
|
|
|
|
|
ENV { |
|
|
accelerator: "gpu" |
|
|
min_memory: "4GB" |
|
|
precision: "fp16" |
|
|
backend: "oktoseek" |
|
|
install_missing: true |
|
|
} |
|
|
|
|
|
DATASET { |
|
|
train: "dataset/train.jsonl" |
|
|
validation: "dataset/val.jsonl" |
|
|
} |
|
|
|
|
|
MODEL { |
|
|
base: "t5-small" |
|
|
device: "auto" |
|
|
} |
|
|
|
|
|
TRAIN { |
|
|
epochs: 3 |
|
|
batch_size: 8 |
|
|
learning_rate: 0.0001 |
|
|
device: "auto" |
|
|
} |
|
|
|
|
|
EXPLORER { |
|
|
try { |
|
|
lr: [0.001, 0.0005, 0.0001] |
|
|
batch_size: [4, 8, 16] |
|
|
optimizer: ["adamw", "sgd"] |
|
|
} |
|
|
max_tests: 5 |
|
|
pick_best_by: "val_loss" |
|
|
} |
|
|
|
|
|
MONITOR { |
|
|
metrics: ["loss", "val_loss", "accuracy"] |
|
|
log_to: "logs/explorer_test.log" |
|
|
} |
|
|
|
|
|
EXPORT { |
|
|
format: ["okm"] |
|
|
path: "export/" |
|
|
} |
|
|
|
|
|
|
