Reduce size of docker image for DeepSpeech 3

Signed-off-by: Beat Buesser <[email protected]>

Author: beat-buesser

.github/actions/deepspeech-v3/Dockerfile

@@ -1,5 +1,5 @@
-pod# Get base from a pytorch image
-FROM pytorch/pytorch:2.1.1-cuda12.1-cudnn8-runtime
+# Get base from a pytorch image
+FROM python:3.10.13-slim-bullseye
 
 # Set to install things in non-interactive mode
 ENV DEBIAN_FRONTEND noninteractive
@@ -26,10 +26,10 @@ RUN apt-get update \
 RUN pip install --ignore-installed PyYAML torch==2.1.1 tensorflow==2.14.1 torchaudio==2.1.1 pytorch-lightning==2.1.2 scikit-learn==1.3.2
 RUN pip install --no-build-isolation fairscale==0.4.13
 
-RUN git clone https://github.com/SeanNaren/deepspeech.pytorch.git
-RUN cd deepspeech.pytorch && sed -i '/^sklearn/d' requirements.txt && pip install -r requirements.txt && pip install -e .
+RUN git clone https://github.com/SeanNaren/deepspeech.pytorch.git && cd deepspeech.pytorch && sed -i '/^sklearn/d' requirements.txt && pip install -r requirements.txt && pip install -e .
 
 RUN pip install numba==0.56.4 pytest-cov==4.1.0 pydub==0.25.1
-RUN pip list
 
 RUN mkdir -p /root/.art/data && cd /root/.art/data && curl -LJO "https://github.com/SeanNaren/deepspeech.pytorch/releases/download/V3.0/librispeech_pretrained_v3.ckpt"
+
+CMD ["/bin/bash"]

tests/estimators/speech_recognition/test_pytorch_deep_speech.json

@@ -241,26 +241,26 @@
         1611.2112
       ],
       "v3": [
-        -524.2962,
-        -260.02936,
-        578.2185,
-        -158.79018,
-        -37.202496,
-        -311.07156,
-        -3.984712,
-        166.36012,
-        -15.264099,
-        199.08426,
-        72.971436,
-        175.62677,
-        -404.36395,
-        -171.99176,
-        370.83533,
-        -220.93425,
-        -1.2995263,
-        146.4521,
-        -381.7052,
-        -69.52321
+        -320.31787,
+        458.37903,
+        166.74754,
+        -60.69311,
+        -110.62456,
+        -219.13577,
+        73.11543,
+        -242.01416,
+        133.31516,
+        112.652115,
+        60.64424,
+        -216.93463,
+        92.05547,
+        89.600334,
+        -168.58362,
+        3.0452464,
+        144.53711,
+        299.63873,
+        124.885254,
+        56.88322
       ]
     }
   },