From d3c6e956a24039ca7c361552858138069406b3b2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jakub=20Zar=C4=99ba?= <k.mjayk@gmail.com>
Date: Wed, 10 May 2023 22:55:38 +0200
Subject: [PATCH] s

---
 mlflow_config.yaml | 7 +++++++
 1 file changed, 7 insertions(+)
 create mode 100644 mlflow_config.yaml

diff --git a/mlflow_config.yaml b/mlflow_config.yaml
new file mode 100644
index 0000000..b13701f
--- /dev/null
+++ b/mlflow_config.yaml
@@ -0,0 +1,7 @@
+name: s487187-training
+
+entry_points:
+  main:
+    parameters:
+      epochs: int
+    command: "python train.py --epochs {epochs}"
\ No newline at end of file