ml_pipeline_cookiecutter/{{cookiecutter.project_slug}}/{{cookiecutter.module_name}}/config/training.toml

9 lines
106 B
TOML
Raw Permalink Normal View History

2024-04-06 13:02:31 -07:00
[training]
batch_size = 16
epochs = 10
learning_rate = 0.01
device = 'cpu'
# examples = 50
examples = -1