11 lines
269 B
Plaintext
11 lines
269 B
Plaintext
|
name: ium_s434766-mlflow
|
||
|
|
|
||
|
|
docker_env:
|
||
|
|
image: owczarczykp/ium_s434766:latest
|
||
|
|
|
||
|
|
entry_points:
|
||
|
|
main:
|
||
|
|
parameters:
|
||
|
|
num_epochs: {type: int, default: 10}
|
||
|
|
batch_size: {type: int, default: 16}
|
||
|
|
command: "python3 lab8-mlflow.py -r {batch_size} {num_epochs}"
|