-
Install the Celeb-A dataset
-
Celeb-A Faces dataset can be downloaded at the linked site, or in Google Drive. The dataset will download as a file named
img_align_celeba.zip
-
Extract the dataset into
data/celeba
-
Install Pytorch, NumPy, Matplotlib
-
Install Pytorch Ignite using
pip:
pip install pytorch-ignite
conda:
conda install ignite -c pytorch
-
CUDA Device(GPU) must be present to run the IPython notebook.
-
Run all cells of the IPython Notebook.
-
Notifications
You must be signed in to change notification settings - Fork 1
Implementation of the Self-Attention GAN research paper in PyTorch.
saksham-koul/Self-Attention-GAN
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
Implementation of the Self-Attention GAN research paper in PyTorch.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published