Skip to content

Update Dockerfile to use v1.0.0 release checkpoint#5

Open
colbyford wants to merge 1 commit intoaqlaboratory:mainfrom
colbyford:main
Open

Update Dockerfile to use v1.0.0 release checkpoint#5
colbyford wants to merge 1 commit intoaqlaboratory:mainfrom
colbyford:main

Conversation

@colbyford
Copy link
Copy Markdown
Contributor

This minor update to the Dockerfile automatically pulls with epoch 40 checkpoint from the v1.0.0 release.

Note that there is a small bug in how the sample_unconditional.py script assembles the path to the check point.

This command:

python genie/sample_unconditional.py --name base --epoch 40 --scale 0.6 --outdir results/base/outputs

...should point to results/base/checkpoints/epoch.40.ckpt, but instead looks for results/base/checkpoints/epoch=40.ckpt.
There is a temporary symbolic link in this Dockerfile that can be removed later when this bug is fixed.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant