This commit is contained in:
Jakub Zaręba 2023-05-10 22:55:38 +02:00
parent f562d1b0a4
commit d3c6e956a2

7
mlflow_config.yaml Normal file
View File

@ -0,0 +1,7 @@
name: s487187-training
entry_points:
main:
parameters:
epochs: int
command: "python train.py --epochs {epochs}"