12 lines
264 B
Plaintext
12 lines
264 B
Plaintext
|
name: 434695-mlflow
|
||
|
|
|
||
|
|
docker_env:
|
||
|
|
image: shroomy/ium:1
|
||
|
|
|
||
|
|
entry_points:
|
||
|
|
main:
|
||
|
|
parameters:
|
||
|
|
epochs: {type: int, default: 15}
|
||
|
|
batch_size: {type: int, default: 16}
|
||
|
|
command: "python3 vgsales-mlflow.py {epochs} {batch_size}"
|