FROM allenlao/pytorch-mt-dnn:v0.1

# Copy modified source code in
# Base commit: a7f74e0afcffd17ab68fb752fa1cc06eabaacda3
COPY ./src /code/mt-dnn
WORKDIR /code/mt-dnn
