Update Dockerfile

Revert "Update train.py"

This reverts commit e1f7a3c131af2ba04fab27d92c270255f34c1f38.

Update train.py

Dockerfile

@@ -21,6 +21,10 @@ ENV PYTHONUNBUFFERED=1
 ENV TQDM_POSITION=-1
 ENV TQDM_MININTERVAL=1
 
+ENV OPENBLAS_NUM_THREADS=1
+ENV MKL_NUM_THREADS=1
+ENV OMP_NUM_THREADS=1
+
 RUN mkdir /.cache && chmod 777 /.cache && chmod 777 /app
 
 CMD ["python3", "train.py"]