File size: 627 Bytes
5fc8c9d |
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 40 41 42 |
PROJECT "VisionClassifier"
DESCRIPTION "Image classification example for computer vision tasks"
DATASET {
train: "dataset/images/"
format: "image+caption"
type: "vision"
augmentation: ["flip", "rotate", "brightness"]
}
MODEL {
base: "oktoseek/vision-base"
architecture: "cnn"
parameters: 50M
precision: "fp16"
}
TRAIN {
epochs: 20
batch_size: 64
learning_rate: 0.001
optimizer: "adam"
device: "cuda"
early_stopping: true
}
METRICS {
accuracy
precision
recall
f1
}
EXPORT {
format: ["onnx", "okm"]
path: "export/"
quantization: "int8"
}
|