FROM pytorch/pytorch:latest
RUN pip install tqdm
COPY mnist.py mnist.py