Commit ·
0222c6b
1
Parent(s): b566e22
Update README.md
Browse files
README.md
CHANGED
|
@@ -15,6 +15,8 @@ Training notebook: https://colab.research.google.com/drive/16o1TdrxnQ54Msbr813Xf
|
|
| 15 |
|
| 16 |
Inference notebook: https://colab.research.google.com/drive/1e7dR2dptM8F1xhRcyy-Aqow9YSe0NE3z?usp=sharing
|
| 17 |
|
|
|
|
|
|
|
| 18 |
Please credit if you use this, and feedback on the code is welcomed :)
|
| 19 |
|
| 20 |
## Demo:
|
|
|
|
| 15 |
|
| 16 |
Inference notebook: https://colab.research.google.com/drive/1e7dR2dptM8F1xhRcyy-Aqow9YSe0NE3z?usp=sharing
|
| 17 |
|
| 18 |
+
The lightwightgan code has an assert requiring a GPU. For inference on the CPU we ned to re-define the Generator class and some other functions - see minimal example here: https://colab.research.google.com/drive/1fnHLdJ7niPMGOOBjGkNsnc6iADpf1Ujd?usp=sharing . This approach was used to make the demo space here: https://huggingface.co/spaces/johnowhitaker/orbgan_demo
|
| 19 |
+
|
| 20 |
Please credit if you use this, and feedback on the code is welcomed :)
|
| 21 |
|
| 22 |
## Demo:
|