14 lines
363 B
Plaintext
14 lines
363 B
Plaintext
|
name: tutorial
|
||
|
|
|
||
|
|
# conda_env: conda.yaml #ścieżka do pliku conda.yaml z definicją środowiska
|
||
|
|
|
||
|
|
docker_env:
|
||
|
|
image: mikolajk/ium:mlflow
|
||
|
|
|
||
|
|
entry_points:
|
||
|
|
main:
|
||
|
|
parameters:
|
||
|
|
numberOfEpochParam: {type: float, default: 3}
|
||
|
|
learning_rate: {type: float, default: 0.1}
|
||
|
|
command: "python ./lab8/trainScript.py {numberOfEpochParam} {learning_rate}"
|