diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..347b3ce --- /dev/null +++ b/.gitignore @@ -0,0 +1,2 @@ +out/ +training-runs/ diff --git a/README.md b/README.md index 6d0e19e..ff4a5e5 100644 --- a/README.md +++ b/README.md @@ -2,6 +2,8 @@ * Included Peter Baylies' projector from [Schultz]'s StyleGAN repo * adapted gen_images.py to allow use from other python code +* Added `kernel.json`. Place this under `~/.local/share/jupyter/kernels/paris-stylegan3/kernel.json` to be able to use the Docker container in JupyterLab. + ## Alias-Free Generative Adversarial Networks (StyleGAN3)
Official PyTorch implementation of the NeurIPS 2021 paper diff --git a/kernel.json b/kernel.json new file mode 100644 index 0000000..a53a501 --- /dev/null +++ b/kernel.json @@ -0,0 +1,29 @@ +{ +"argv": [ + "/usr/bin/docker", + "run", + "--shm-size=2g", + "--gpus", + "all", + "--network=host", + "--rm", + "-v", + "/home/ruben/paris/stylegan3:/scratch", + "--user", + "1000:1000", + "--workdir=/scratch", + "-e", + "HOME=/scratch", + "-v", + "{connection_file}:/connection-spec", + "stylegan3", + "python", + "-m", + "ipykernel_launcher", + "-f", + "/connection-spec" +], +"display_name": "paris-stylegan3", +"language":"python" +} +