From 0d403e9619749f1721fa7ec29e9e8d3a9a5e4c3a Mon Sep 17 00:00:00 2001 From: "Colby T. Ford" Date: Sun, 23 Jun 2024 13:35:14 -0700 Subject: [PATCH] Update Dockerfile to use v1.0.0 release checkpoint --- Dockerfile | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/Dockerfile b/Dockerfile index 75fa00fb..55bf58e3 100644 --- a/Dockerfile +++ b/Dockerfile @@ -30,6 +30,11 @@ RUN git clone https://github.com/aqlaboratory/genie2.git \ WORKDIR /app/genie2/ -## Install the base checkpoint -RUN cd results/base \ - && git lfs pull +## Download the base checkpoint (from v1.0.0 release) +RUN mkdir results/base/checkpoints \ + && cd results/base/checkpoints \ + # && git lfs pull + # && wget https://github.com/aqlaboratory/genie2/releases/download/v1.0.0/epoch.30.ckpt \ + && wget https://github.com/aqlaboratory/genie2/releases/download/v1.0.0/epoch.40.ckpt \ + ## Hack to make the checkpoint loadable + && ln -s epoch.40.ckpt epoch=40.ckpt