sieve.Model
A sieve.Model
is a Python class that defines a model in Sieve. A model is a function that has a __setup__
and a __predict__
step. The __setup__
step is run once when the model initializes, and the __predict__
step is run every time the model is called.
YOLO Object Detection Example
Arguments
name
Name of the function.gpu
Whether the function should run on a GPU. Defaults toFalse
.python_packages
List of Python packages to install. Defaults to[]
.python_version
Python version to use. Defaults to"3.8"
.system_packages
List of system packages to install. Defaults to[]
.cuda_version
CUDA version to use. Defaults toNone
.machine_type
Machine type to use. Defaults toNone
.iterator_input
Whether the function takes in an iterator. Defaults toFalse
.run_commands
List of commands to run when the function is built. Defaults to[]
. The run commands are run when the Docker container is built, and the resulting files are a part of the Docker container.
name
Name of the function.
gpu
Whether the function should run on a GPU.
python_packages
List of Python packages to install.
python_version
Python version to use.
system_packages
List of system packages to install.
cuda_version
CUDA version to use.
machine_type
Machine type to use.
iterator_input
Whether the function takes in an iterator.
run_commands
List of commands to run when the function is built.
Was this page helpful?