From 6ef5e3fc7e3af2a9bc7d3bbb08d8e394f68b2124 Mon Sep 17 00:00:00 2001
From: Jakub Henyk <jakubhenyk@gmail.com>
Date: Sun, 7 May 2023 18:39:46 +0200
Subject: [PATCH] fix17

---
 Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index edd189c..9f9fc51 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -28,5 +28,5 @@ RUN chmod +x ./train.py
 ARG epochs=5
 RUN echo $epochs
 
-CMD python3 ./train.py epochs
+CMD python3 ./train.py $epochs
 #CMD python3 ./test.py
\ No newline at end of file