jupyter kernel definition

.gitignore

@@ -0,0 +1,2 @@
+out/
+training-runs/

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)<br><sub>Official PyTorch implementation of the NeurIPS 2021 paper</sub>

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"
+}
+