1 2 3 4 5
FROM python:3.9-slim WORKDIR /app COPY training/ /app/ RUN pip install --no-cache-dir -q "numpy>=1.24.0" CMD ["python3", "train_mnist_e2e.py", "--epochs", "2"]