10 lines
156 B
Docker
10 lines
156 B
Docker
ARG VERSION
|
|
FROM tensorflow/tensorflow:$VERSION
|
|
WORKDIR /
|
|
|
|
RUN pip install keras h5py
|
|
|
|
COPY /trainer /trainer
|
|
|
|
ENTRYPOINT ["python", "-m", "trainer.start"]
|