From 3d615bca61fb194e3c5e9f0a7355d29db1c5e5f3 Mon Sep 17 00:00:00 2001 From: Aladdin Persson Date: Wed, 24 Mar 2021 21:09:02 +0100 Subject: [PATCH] update progan readme linkW --- ML/Pytorch/GANs/ProGAN/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ML/Pytorch/GANs/ProGAN/README.md b/ML/Pytorch/GANs/ProGAN/README.md index 18a6eb5..1c25f38 100644 --- a/ML/Pytorch/GANs/ProGAN/README.md +++ b/ML/Pytorch/GANs/ProGAN/README.md @@ -14,7 +14,7 @@ The model was trained on the Celeb-HQ dataset up to 256x256 image size. After th The dataset can be downloaded from Kaggle: [link](https://www.kaggle.com/lamsimon/celebahq). ### Download pretrained weights -Pretrained weights [here](https://github.com/aladdinpersson/Machine-Learning-Collection/releases/download/1.0/ProGAN_weights.zip). +Download pretrained weights [here](https://github.com/aladdinpersson/Machine-Learning-Collection/releases/download/1.0/ProGAN_weights.zip). Extract the zip file and put the pth.tar files in the directory with all the python files. Make sure you put LOAD_MODEL=True in the config.py file.