2021-05-16 18:53:33 +02:00
|
|
|
name: 426206mlflow
|
|
|
|
|
|
|
|
|
|
#conda_env: conda.yaml #ścieżka do pliku conda.yaml z definicją środowiska
|
|
|
|
|
|
2021-05-16 19:55:22 +02:00
|
|
|
docker_env:
|
|
|
|
|
image: rokoch/ium:01
|
2021-05-16 18:53:33 +02:00
|
|
|
|
|
|
|
|
entry_points:
|
|
|
|
|
main:
|
|
|
|
|
parameters:
|
|
|
|
|
epochs: {type: int, default: 100}
|
|
|
|
|
lr: {type: float, default: 0.001}
|
|
|
|
|
command: "python3 train_mlflow.py -e {epochs} -l {lr}"
|
